Update neural_lam/datastore/base.py

Co-authored-by: Joel Oskarsson <[email protected]>
mllam · Sep 10, 2024 · 5bda935 · 5bda935
1 parent 90bc594
commit 5bda935
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_lam/datastore/base.py b/neural_lam/datastore/base.py
@@ -121,7 +121,7 @@ def get_num_data_vars(self, category: str) -> int:
         Parameters
         ----------
         category : str
-            The category of the variables.
+            The category of the variables (state/forcing/static).
 
         Returns
         -------