fix another import error

SWIFTSIM · Feb 23, 2024 · c3ddc86 · c3ddc86
1 parent 692d55b
commit c3ddc86
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/swiftsimio/visualisation/slice_backends/nearest_neighbors.py b/swiftsimio/visualisation/slice_backends/nearest_neighbors.py
@@ -1,4 +1,4 @@
-from numpy import float64, float32, ndarray, linspace, array, stack, meshgrid
+from numpy import float64, float32, ndarray, linspace, array, stack, meshgrid, power
 
 from swiftsimio import SWIFTDataset, cosmo_array
 from swiftsimio.optional_packages import KDTree, TREE_AVAILABLE
@@ -117,13 +117,13 @@ def get_hsml(data: SWIFTDataset) -> cosmo_array:
     The extracted "smoothing lengths".
     """
     try:
-        hsml = np.power(data.gas.volume, 1.0 / 3.0)
+        hsml = power(data.gas.volume, 1.0 / 3.0)
     except AttributeError:
         try:
             # Try computing the volumes explicitly?
             masses = data.gas.masses
             densities = data.gas.densities
-            hsml = np.power(masses / densities, 1.0 / 3.0)
+            hsml = power(masses / densities, 1.0 / 3.0)
         except AttributeError:
             # Fall back to SPH behavior if above didn't work...
             hsml = sph_get_hsml(data)