Check import of rasterio sort_xy function

CartoDB · Oct 31, 2024 · 135ae31 · 135ae31
1 parent 5778b4c
commit 135ae31
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/raster_loader/io/common.py b/raster_loader/io/common.py
@@ -13,7 +13,6 @@
 import rio_cogeo
 import rasterio
 import quadbin
-from rasterio.sample import sort_xy
 
 from raster_loader.geo import raster_bounds
 from raster_loader.errors import (
@@ -404,7 +403,12 @@ def not_enough_samples():
         x = rng.uniform(west, east, n_samples)
         y = rng.uniform(south, north, n_samples)
 
-        coords = sort_xy(zip(x, y))
+        try:
+            from rasterio.sample import sort_xy
+        except ImportError:
+            coords = zip(x, y)
+        else:
+            coords = sort_xy(zip(x, y))
 
         samples = raster_dataset.sample(coords, indexes=bands, masked=True)
         for sample in samples: