Merge pull request #316 from OpenFreeEnergy/bugfix-settingsbasemodel

Switch to matching against `SettingsBaseModel` instead of `Settings` in `gufe_to_subgraph`
OpenFreeEnergy · Oct 29, 2024 · 63c1bb9 · 63c1bb9
2 parents 765d2b7 + 6080a9f
commit 63c1bb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alchemiscale/storage/statestore.py b/alchemiscale/storage/statestore.py
@@ -18,9 +18,9 @@
     AlchemicalNetwork,
     Transformation,
     NonTransformation,
-    Settings,
     Protocol,
 )
+from gufe.settings import SettingsBaseModel
 from gufe.tokenization import GufeTokenizable, GufeKey, JSON_HANDLER
 
 from neo4j import Transaction, GraphDatabase, Driver
@@ -342,7 +342,7 @@ def _gufe_to_subgraph(
                 ):
                     node[key] = json.dumps(value, cls=JSON_HANDLER.encoder)
                     node["_json_props"].append(key)
-            elif isinstance(value, Settings):
+            elif isinstance(value, SettingsBaseModel):
                 node[key] = json.dumps(value, cls=JSON_HANDLER.encoder, sort_keys=True)
                 node["_json_props"].append(key)
             elif isinstance(value, GufeTokenizable):