Use standard dict instead of defaultdict

OpenFreeEnergy · Jan 22, 2025 · 4056752 · 4056752
1 parent 7f546d0
commit 4056752
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/alchemiscale/storage/statestore.py b/alchemiscale/storage/statestore.py
@@ -1238,14 +1238,14 @@ def _node_to_scoped_key(node):
             return ScopedKey.from_str(node["_scoped_key"])
 
         transform_function = _node_to_gufe if return_gufe else _node_to_scoped_key
-        transform_results = defaultdict(None)
+        transform_results = {}
         for record in query_results.records:
             node = record_data_to_node(record["th"])
             network_scoped_key = record["an"]["_scoped_key"]
             transform_results[network_scoped_key] = transform_function(node)
 
         return [
-            transform_results[str(network_scoped_key)]
+            transform_results.get(str(network_scoped_key))
             for network_scoped_key in network_scoped_keys
         ]