try putting __base__ back in

materialsproject · Nov 25, 2024 · 56ba715 · 56ba715
1 parent ef2fa73
commit 56ba715
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mp_api/client/core/client.py b/mp_api/client/core/client.py
@@ -1070,6 +1070,7 @@ def _generate_returned_model(self, doc):
             "MPDataDoc",
             fields_not_requested=(list[str], unset_fields),
             **include_fields,
+            __base__=self.document_model,
         )
 
         def new_repr(self) -> str: