From d4d438ff372ce6568c62717c4bd857de83f888a5 Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Fri, 27 Sep 2024 17:59:43 +0200
Subject: [PATCH] add missing kwarg to HiLAM.__init__

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

diff --git a/neural_lam/models/hi_lam.py b/neural_lam/models/hi_lam.py
index ec05d59c..68ed0e6e 100644
--- a/neural_lam/models/hi_lam.py
+++ b/neural_lam/models/hi_lam.py
@@ -13,8 +13,8 @@ class HiLAM(BaseHiGraphModel):
     The Hi-LAM model from Oskarsson et al. (2023)
     """
 
-    def __init__(self, args):
-        super().__init__(args)
+    def __init__(self, args, datastore):
+        super().__init__(args, datastore=datastore)
 
         # Make down GNNs, both for down edges and same level
         self.mesh_down_gnns = nn.ModuleList(