Fix docstring

microsoft · Aug 15, 2024 · 7a32564 · 7a32564
1 parent 12d6db9
commit 7a32564
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/aurora/batch.py b/aurora/batch.py
@@ -50,11 +50,10 @@ class Batch:
 
     Args:
         surf_vars (dict[str, :class:`torch.Tensor`]): Surface-level variables with shape
-            `(B, T, H, W)`.
-        static_vars (dict[str, :class:`torch.Tensor`]): Static variables with shape
-            `(B, T, H, W)`.
+            `(b, t, h, w)`.
+        static_vars (dict[str, :class:`torch.Tensor`]): Static variables with shape `(h, w)`.
         atmos_vars (dict[str, :class:`torch.Tensor`]): Atmospheric variables with shape
-            `(B, T, C, H, W)`.
+            `(b, t, c, h, w)`.
         metadata (:class:`Metadata`): Metadata associated to this batch.
     """