Update neural_lam/datastore/base.py

Co-authored-by: Joel Oskarsson <joel.oskarsson@outlook.com>
mllam · Sep 10, 2024 · bcaa919 · bcaa919
1 parent 4fc2448
commit bcaa919
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_lam/datastore/base.py b/neural_lam/datastore/base.py
@@ -87,7 +87,7 @@ def get_vars_units(self, category: str) -> List[str]:
         Parameters
         ----------
         category : str
-            The category of the variables.
+            The category of the variables (state/forcing/static).
 
         Returns
         -------