migrate xarray to rioxarray

IMMM-SFA · Mar 21, 2024 · 255ab4e · 255ab4e
1 parent 7d032f5
commit 255ab4e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/cerf/utils.py b/cerf/utils.py
@@ -4,7 +4,7 @@
 
 import numpy as np
 import pandas as pd
-import rasterio
+import rioxarray
 import xarray as xr
 import geopandas as gpd
 from shapely.geometry import Point
@@ -253,7 +253,7 @@ def raster_to_coord_arrays(template_raster):
     """
 
     # Read the data
-    da = xr.open_dataset(template_raster, engine="rasterio")
+    da = rioxarray.open_rasterio(template_raster)
 
     # Compute the lon/lat coordinates with rasterio.warp.transform
     x, y = np.meshgrid(da['x'], da['y'])

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 numpy>=1.19.4
 pandas>=1.1.4
 rasterio>=1.2.3
-xarray>=0.16.1
+rioxarray>=0.15.1
 PyYAML>=5.4.1
 requests>=2.25.1
 joblib>=1.0.1

diff --git a/setup.py b/setup.py
@@ -34,7 +34,7 @@ def get_requirements():
         'numpy>=1.19.4',
         'pandas>=1.1.4',
         'rasterio>=1.2.3',
-        'xarray>=0.16.1',
+        'rioxarray>=0.15.1',
         'PyYAML>=5.4.1',
         'requests>=2.25.1',
         'joblib>=1.0.1',