From 5bda935e8934761a1b307399e4a5772aae332c3e Mon Sep 17 00:00:00 2001
From: Leif Denby <leif@denby.eu>
Date: Tue, 10 Sep 2024 17:19:18 +0200
Subject: [PATCH] Update neural_lam/datastore/base.py

Co-authored-by: Joel Oskarsson <joel.oskarsson@outlook.com>
---
 neural_lam/datastore/base.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/neural_lam/datastore/base.py b/neural_lam/datastore/base.py
index 66c18670..b8c7afa8 100644
--- 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
         -------