Fix bug in datastore loading in graph creation script

mllam · Jan 10, 2025 · b296095 · b296095
1 parent 5d94325
commit b296095
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_lam/build_rectangular_graph.py b/neural_lam/build_rectangular_graph.py
@@ -9,7 +9,7 @@
 
 # Local
 from . import utils
-from .config import load_config_and_datastore
+from .config import load_config_and_datastores
 
 WMG_ARCHETYPES = {
     "keisler": wmg.create.archetype.create_keisler_graph,
@@ -81,7 +81,7 @@ def main(input_args=None):
         args.graph_name is not None
     ), "Specify the name to save graph as with --graph_name"
 
-    _, datastore, datastore_boundary = load_config_and_datastore(
+    _, datastore, datastore_boundary = load_config_and_datastores(
         config_path=args.config_path
     )