xarray testing still in progress

Author: browniea

mhkit/river/io/d3d.py

@@ -16,15 +16,15 @@
 from mhkit.utils import unorm
 
 
-def get_all_time(data: netCDF4.Dataset) -> NDArray:
+def get_all_time(data: (netCDF4.Dataset, xr.Dataset)) -> NDArray:
     """
     Returns all of the time stamps from a D3D simulation passed to the function
     as a NetCDF object (data)
 
     Parameters
     ----------
-    data: NetCDF4 object
-       A NetCDF4 object that contains spatial data, e.g. velocity or shear
+    data: NetCDF4 object or xarray Dataset
+       A NetCDF4 object or xarray Dataset that contains spatial data, e.g. velocity or shear
        stress generated by running a Delft3D model.
 
     Returns