Merge pull request #4656 from neutrinoceros/unyt_3.0_compat_3

yt-project · Sep 1, 2023 · 196defa · 196defa
2 parents db6862e + 6ffca39
commit 196defa
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/yt/frontends/halo_catalog/io.py b/yt/frontends/halo_catalog/io.py
@@ -4,7 +4,7 @@
 
 from yt.frontends.gadget_fof.io import IOHandlerGadgetFOFHaloHDF5
 from yt.funcs import parse_h5_attr
-from yt.units.yt_array import uvstack  # type: ignore
+from yt.units._numpy_wrapper_functions import uvstack
 from yt.utilities.io_handler import BaseParticleIOHandler
 from yt.utilities.on_demand_imports import _h5py as h5py
 

diff --git a/yt/frontends/ytdata/io.py b/yt/frontends/ytdata/io.py
@@ -2,7 +2,7 @@
 
 from yt.funcs import mylog, parse_h5_attr
 from yt.geometry.selection_routines import GridSelector
-from yt.units.yt_array import uvstack  # type: ignore
+from yt.units._numpy_wrapper_functions import uvstack
 from yt.utilities.io_handler import BaseIOHandler
 from yt.utilities.on_demand_imports import _h5py as h5py
 

diff --git a/yt/visualization/plot_modifications.py b/yt/visualization/plot_modifications.py
@@ -21,7 +21,8 @@
 from yt.geometry.api import Geometry
 from yt.geometry.unstructured_mesh_handler import UnstructuredIndex
 from yt.units import dimensions
-from yt.units.yt_array import YTArray, YTQuantity, uhstack  # type: ignore
+from yt.units._numpy_wrapper_functions import uhstack
+from yt.units.yt_array import YTArray, YTQuantity
 from yt.utilities.exceptions import (
     YTDataTypeUnsupported,
     YTFieldNotFound,