add xarray to pyproject.toml

Author: mjreno

flopy/discretization/structuredgrid.py

@@ -1676,7 +1676,8 @@ def get_plottable_layer_array(self, a, layer):
         return plotarray
 
     def dataset(self, modeltime=None, mesh=None):
-        import xarray as xr
+        from ..utils import import_optional_dependency
+        xr = import_optional_dependency("xarray")
 
         if modeltime is None:
             raise ValueError("modeltime required for dataset timeseries")

flopy/discretization/vertexgrid.py

@@ -601,7 +601,8 @@ def get_plottable_layer_array(self, a, layer):
         return plotarray
 
     def dataset(self, modeltime=None, mesh=None):
-        import xarray as xr
+        from ..utils import import_optional_dependency
+        xr = import_optional_dependency("xarray")
 
         FILLNA_INT32 = np.int32(-2147483647)
         FILLNA_DBL = 9.96920996838687e36

pyproject.toml

@@ -84,6 +84,7 @@ optional = [
     "vtk >=9.4.0",
     "xmipy",
     "h5py",
+    "xarray",
 ]
 doc = [
     "flopy[optional]",