try model_copy

materialsproject · Nov 25, 2024 · d526bc6 · d526bc6
1 parent c16c1eb
commit d526bc6
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/mp_api/client/core/client.py b/mp_api/client/core/client.py
@@ -1071,14 +1071,16 @@ def _generate_returned_model(self, doc):
             __base__=self.document_model,  # type: ignore
         )
 
-        data_model.model_fields = {
-            **{
-                name: description
-                for name, description in data_model.model_fields.items()
-                if name in set_fields
-            },
-            "fields_not_requested": data_model.model_fields["fields_not_requested"],
-        }
+        data_model = data_model.model_copy(
+            update={
+                **{
+                    name: description
+                    for name, description in data_model.model_fields.items()
+                    if name in set_fields
+                },
+                "fields_not_requested": data_model.model_fields["fields_not_requested"],
+            }
+        )
 
         def new_repr(self) -> str:
             extra = ",\n".join(