Draft of example notebook

microsoft · Aug 12, 2024 · 4c7c3d8 · 4c7c3d8
1 parent 18a450a
commit 4c7c3d8
Show file tree

Hide file tree

Showing 2 changed files with 309 additions and 8 deletions.
diff --git a/aurora/batch.py b/aurora/batch.py
@@ -104,7 +104,12 @@ def crop(self, patch_size: int) -> "Batch":
                 surf_vars={k: v[..., :-1, :] for k, v in self.surf_vars.items()},
                 static_vars={k: v[..., :-1, :] for k, v in self.static_vars.items()},
                 atmos_vars={k: v[..., :-1, :] for k, v in self.atmos_vars.items()},
-                metadata=self.metadata,
+                metadata=Metadata(
+                    lat=self.metadata.lat[:-1],
+                    lon=self.metadata.lon,
+                    atmos_levels=self.metadata.atmos_levels,
+                    time=self.metadata.time,
+                ),
             )
         else:
             raise ValueError(

diff --git a/docs/example.ipynb b/docs/example.ipynb