add missing kwarg to HiLAMParallel

mllam · Sep 27, 2024 · 1889771 · 1889771
1 parent d4d438f
commit 1889771
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neural_lam/models/hi_lam_parallel.py b/neural_lam/models/hi_lam_parallel.py
@@ -16,8 +16,8 @@ class HiLAMParallel(BaseHiGraphModel):
     of Hi-LAM.
     """
 
-    def __init__(self, args):
-        super().__init__(args)
+    def __init__(self, args, datastore):
+        super().__init__(args, datastore=datastore)
 
         # Processor GNNs
         # Create the complete edge_index combining all edges for processing