lazy import for xarray

microsoft · Jan 9, 2025 · 5d7b27a · 5d7b27a
1 parent 871d080
commit 5d7b27a
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/requirements/datasets.txt b/requirements/datasets.txt
@@ -7,3 +7,5 @@ pycocotools==2.0.8
 pyvista==0.44.2
 scikit-image==0.25.0
 scipy==1.14.1
+xarray==2024.11.0
+netcdf4==1.7.2
diff --git a/tests/datasets/test_glacier_mapping_alps.py b/tests/datasets/test_glacier_mapping_alps.py
@@ -13,6 +13,9 @@
 
 from torchgeo.datasets import DatasetNotFoundError, GlacierMappingAlps
 
+pytest.importorskip('xarray', minversion='2024.11.0')
+pytest.importorskip('netCDF4', minversion='1.7.2')
+
 
 class TestGlacierMappingAlps:
     @pytest.fixture(

diff --git a/torchgeo/datasets/glacier_mapping_alps.py b/torchgeo/datasets/glacier_mapping_alps.py
@@ -12,13 +12,17 @@
 import numpy as np
 import pandas as pd
 import torch
-import xarray as xr
 from matplotlib.figure import Figure
 from torch import Tensor
 
 from .errors import DatasetNotFoundError
 from .geo import NonGeoDataset
-from .utils import download_and_extract_archive, download_url, extract_archive
+from .utils import (
+    download_and_extract_archive,
+    download_url,
+    extract_archive,
+    lazy_import,
+)
 
 
 class GlacierMappingAlps(NonGeoDataset):
@@ -220,6 +224,7 @@ def __getitem__(self, index: int) -> dict[str, Tensor]:
                 * the cloud and shadow mask
                 * the additional features (DEM, derived features, etc.) if required
         """
+        xr = lazy_import('xarray')
         nc = xr.open_dataset(
             self.fp_patches[index], decode_coords='all', mask_and_scale=True
         )