Merge pull request #24 from ClimateBenchPress/basepath_args

Make basepath a command line argument

Author: treigerm

src/climatebenchpress/data_loader/datasets/cams.py

@@ -2,6 +2,7 @@
 
 import logging
 from pathlib import Path
+import argparse
 
 import xarray as xr
 
@@ -46,8 +47,16 @@ def open(download_path: Path) -> xr.Dataset:
 
 
 if __name__ == "__main__":
-    ds = open_downloaded_canonicalized_dataset(CamsNitrogenDioxideDataset)
-    open_downloaded_tiny_canonicalized_dataset(CamsNitrogenDioxideDataset)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--basepath", type=Path, default=Path())
+    args = parser.parse_args()
+
+    ds = open_downloaded_canonicalized_dataset(
+        CamsNitrogenDioxideDataset, basepath=args.basepath
+    )
+    open_downloaded_tiny_canonicalized_dataset(
+        CamsNitrogenDioxideDataset, basepath=args.basepath
+    )
 
     for v, da in ds.items():
         print(f"- {v}: {da.dims}")

src/climatebenchpress/data_loader/datasets/cmip6/access_atmos.py

@@ -1,6 +1,7 @@
 __all__ = ["Cmip6AtmosphereAccessDataset"]
 
 from pathlib import Path
+import argparse
 
 from ... import (
     open_downloaded_canonicalized_dataset,
@@ -28,8 +29,16 @@ def download(download_path: Path, progress: bool = True):
 
 
 if __name__ == "__main__":
-    ds = open_downloaded_canonicalized_dataset(Cmip6AtmosphereAccessDataset)
-    open_downloaded_tiny_canonicalized_dataset(Cmip6AtmosphereAccessDataset)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--basepath", type=Path, default=Path())
+    args = parser.parse_args()
+
+    ds = open_downloaded_canonicalized_dataset(
+        Cmip6AtmosphereAccessDataset, basepath=args.basepath
+    )
+    open_downloaded_tiny_canonicalized_dataset(
+        Cmip6AtmosphereAccessDataset, basepath=args.basepath
+    )
 
     for v, da in ds.items():
         print(f"- {v}: {da.dims}")

src/climatebenchpress/data_loader/datasets/cmip6/access_ocean.py

@@ -1,6 +1,7 @@
 __all__ = ["Cmip6OceanAccessDataset"]
 
 from pathlib import Path
+import argparse
 
 from ... import (
     open_downloaded_canonicalized_dataset,
@@ -30,8 +31,16 @@ def download(download_path: Path, progress: bool = True):
 
 
 if __name__ == "__main__":
-    ds = open_downloaded_canonicalized_dataset(Cmip6OceanAccessDataset)
-    open_downloaded_tiny_canonicalized_dataset(Cmip6OceanAccessDataset)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--basepath", type=Path, default=Path())
+    args = parser.parse_args()
+
+    ds = open_downloaded_canonicalized_dataset(
+        Cmip6OceanAccessDataset, basepath=args.basepath
+    )
+    open_downloaded_tiny_canonicalized_dataset(
+        Cmip6OceanAccessDataset, basepath=args.basepath
+    )
 
     for v, da in ds.items():
         print(f"- {v}: {da.dims}")

src/climatebenchpress/data_loader/datasets/era5.py

@@ -1,6 +1,7 @@
 __all__ = ["Era5Dataset"]
 
 from pathlib import Path
+import argparse
 
 import xarray as xr
 
@@ -49,8 +50,12 @@ def open(download_path: Path) -> xr.Dataset:
 
 
 if __name__ == "__main__":
-    ds = open_downloaded_canonicalized_dataset(Era5Dataset)
-    open_downloaded_tiny_canonicalized_dataset(Era5Dataset)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--basepath", type=Path, default=Path())
+    args = parser.parse_args()
+
+    ds = open_downloaded_canonicalized_dataset(Era5Dataset, basepath=args.basepath)
+    open_downloaded_tiny_canonicalized_dataset(Era5Dataset, basepath=args.basepath)
 
     for v, da in ds.items():
         print(f"- {v}: {da.dims}")

src/climatebenchpress/data_loader/datasets/esa_biomass_cci.py

@@ -2,6 +2,7 @@
 
 import logging
 from pathlib import Path
+import argparse
 
 import xarray as xr
 
@@ -55,7 +56,13 @@ def open(download_path: Path) -> xr.Dataset:
 
 
 if __name__ == "__main__":
-    ds = open_downloaded_canonicalized_dataset(EsaBiomassCciDataset)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--basepath", type=Path, default=Path())
+    args = parser.parse_args()
+
+    ds = open_downloaded_canonicalized_dataset(
+        EsaBiomassCciDataset, basepath=args.basepath
+    )
     num_lon, num_lat = ds.lon.size, ds.lat.size
     open_downloaded_tiny_canonicalized_dataset(
         EsaBiomassCciDataset,
@@ -64,6 +71,7 @@ def open(download_path: Path) -> xr.Dataset:
             "X": slice(num_lon // 2, (num_lon // 2) + 500),
             "Y": slice(num_lat // 2, (num_lat // 2) + 500),
         },
+        basepath=args.basepath,
     )
 
     for v, da in ds.items():

src/climatebenchpress/data_loader/datasets/nextgems.py

@@ -1,6 +1,7 @@
 __all__ = ["NextGemsDataset"]
 
 from pathlib import Path
+import argparse
 
 import healpy
 import intake
@@ -94,8 +95,12 @@ def _get_nn_lon_lat_index(nside, lons, lats):
 
 
 if __name__ == "__main__":
-    ds = open_downloaded_canonicalized_dataset(NextGemsDataset)
-    open_downloaded_tiny_canonicalized_dataset(NextGemsDataset)
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--basepath", type=Path, default=Path())
+    args = parser.parse_args()
+
+    ds = open_downloaded_canonicalized_dataset(NextGemsDataset, basepath=args.basepath)
+    open_downloaded_tiny_canonicalized_dataset(NextGemsDataset, basepath=args.basepath)
 
     for v, da in ds.items():
         print(f"- {v}: {da.dims}")