Duck-type serialize by default

openforcefield · Jun 6, 2024 · bbc2416 · bbc2416
1 parent 1b597f7
commit bbc2416
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/openff/models/models.py b/openff/models/models.py
@@ -1,3 +1,5 @@
+from typing import Any
+
 from openff.units import Quantity
 
 from openff.models._pydantic import BaseModel
@@ -14,3 +16,9 @@ class DefaultModel(BaseModel):
         "validate_assignment": True,
         "arbitrary_types_allowed": True,
     }
+
+    def model_dump(self, **kwargs) -> dict[str, Any]:
+        return super().model_dump(serialize_as_any=True, **kwargs)
+
+    def model_dump_json(self, **kwargs) -> str:
+        return super().model_dump_json(serialize_as_any=True, **kwargs)
diff --git a/openff/models/validators.py b/openff/models/validators.py
@@ -60,7 +60,9 @@ def coerce_json_back_to_quantity(
             try:
                 return Quantity(*json_loader(v).values())
             except Exception as error:
-                raise ValueError("str types need to be of the structure that can be imported back from JSON") from error
+                raise ValueError(
+                    "str types need to be of the structure that can be imported back from JSON"
+                ) from error
 
         else:
             raise ValueError(