From 1889771efcc04ce304988e54045950fe4cc46e29 Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Fri, 27 Sep 2024 18:01:00 +0200
Subject: [PATCH] add missing kwarg to HiLAMParallel

---
 neural_lam/models/hi_lam_parallel.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/neural_lam/models/hi_lam_parallel.py b/neural_lam/models/hi_lam_parallel.py
index 80181ec0..5c1872ab 100644
--- 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