diff --git a/xradar/io/backends/metek.py b/xradar/io/backends/metek.py index c2d17e54..a570187f 100644 --- a/xradar/io/backends/metek.py +++ b/xradar/io/backends/metek.py @@ -21,7 +21,6 @@ import numpy as np import xarray as xr -from datatree import DataTree from xarray.backends.common import AbstractDataStore, BackendArray, BackendEntrypoint from xarray.backends.file_manager import CachingFileManager from xarray.backends.store import StoreBackendEntrypoint @@ -615,7 +614,7 @@ def open_dataset( def open_metek_datatree(filename_or_obj, **kwargs): - """Open Metek MRR2 dataset as :py:class:`datatree.DataTree`. + """Open Metek MRR2 dataset as :py:class:`xarray.DataTree`. Parameters ---------- @@ -644,7 +643,7 @@ def open_metek_datatree(filename_or_obj, **kwargs): Returns ------- - dtree: datatree.DataTree + dtree: xarray.DataTree DataTree """ # handle kwargs, extract first_dim @@ -674,6 +673,6 @@ def open_metek_datatree(filename_or_obj, **kwargs): ds.insert(0, xr.Dataset()) # open_dataset(filename_or_obj, group="/")) # create datatree root node with required data - dtree = DataTree(data=_assign_root(ds), name="root") + dtree = xr.DataTree(dataset=_assign_root(ds), name="root") # return datatree with attached sweep child nodes return _attach_sweep_groups(dtree, ds[1:])