Merge pull request #869 from materialsproject/bugfix/generic_model_re…

…ference Remove generic model reference
materialsproject · Oct 9, 2023 · 705718f · 705718f
2 parents 8546fed + a8cf989
commit 705718f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/maggma/api/models.py b/src/maggma/api/models.py
@@ -2,7 +2,6 @@
 from typing import Generic, List, Optional, TypeVar
 
 from pydantic import BaseModel, Field, validator
-from pydantic.generics import GenericModel
 
 from maggma import __version__
 
@@ -47,7 +46,7 @@ def from_traceback(cls, traceback):
         pass
 
 
-class Response(GenericModel, Generic[DataT]):
+class Response(BaseModel, Generic[DataT]):
     """
     A Generic API Response
     """