get datastore static data in ARModel with defining split

mllam · Nov 12, 2024 · f2c3a27 · f2c3a27
1 parent b7a10ef
commit f2c3a27
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/neural_lam/models/ar_model.py b/neural_lam/models/ar_model.py
@@ -34,12 +34,10 @@ def __init__(
         self.save_hyperparameters(ignore=["datastore"])
         self.args = args
         self._datastore = datastore
-        # XXX: should be this be somewhere else?
-        split = "train"
         num_state_vars = datastore.get_num_data_vars(category="state")
         num_forcing_vars = datastore.get_num_data_vars(category="forcing")
         da_static_features = datastore.get_dataarray(
-            category="static", split=split
+            category="static", split=None
         )
         da_state_stats = datastore.get_standardization_dataarray(
             category="state"