From 5df2ecf3e217f6513fa15d417ccdc538846b30e5 Mon Sep 17 00:00:00 2001 From: Leif Denby Date: Fri, 27 Sep 2024 17:58:21 +0200 Subject: [PATCH] add missing kwarg to BaseHiGraphModel.__init__ --- neural_lam/models/base_hi_graph_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/neural_lam/models/base_hi_graph_model.py b/neural_lam/models/base_hi_graph_model.py index a2ebcc1b..a81df88f 100644 --- a/neural_lam/models/base_hi_graph_model.py +++ b/neural_lam/models/base_hi_graph_model.py @@ -12,8 +12,8 @@ class BaseHiGraphModel(BaseGraphModel): Base class for hierarchical graph models. """ - def __init__(self, args): - super().__init__(args) + def __init__(self, args, datastore): + super().__init__(args, datastore=datastore) # Track number of nodes, edges on each level # Flatten lists for efficient embedding