feat: Add --load/--no-load

Fixes #32
oceanmodeling · Aug 30, 2024 · 54e30d5 · 54e30d5
1 parent f089458
commit 54e30d5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/inspectds/cli.py b/inspectds/cli.py
@@ -190,6 +190,7 @@ def inspect_dataset(
     variable_attributes: bool = typer.Option(default=False, help="Whether to include the variable attributes in the output"),
     global_attributes: bool = typer.Option(default=False, help="Whether to include the global attributes in the output"),
     full: bool = typer.Option(default=False, help="Display full output. Overrides any other option"),
+    load: bool = typer.Option(default=False, help="Load the data from disk. Only use on small datasets"),
     version: bool = typer.Option(False, "--version", help="Display the version", callback=version_callback, is_eager=True),
 ) -> int:  # fmt: skip
     import xarray as xr
@@ -227,6 +228,9 @@ def inspect_dataset(
     for coord in ds.coords:
         ds[coord].load()
 
+    if load:
+        ds = ds.load()
+
     if full:
         dimensions = coordinates = variables = variable_attributes = global_attributes = True