Merge branch 'feat/datastores' of https://github.com/leifdenby/neural…

…-lam into feat/datastores
mllam · Sep 10, 2024 · c415008 · c415008
2 parents 6998683 + 5bda935
commit c415008
Showing 1 changed file with 3 additions and 3 deletions.
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
         -------
@@ -104,7 +104,7 @@ def get_vars_names(self, category: str) -> List[str]:
         Parameters
         ----------
         category : str
-            The category of the variables.
+            The category of the variables (state/forcing/static).
 
         Returns
         -------
@@ -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
         -------