Import dpnp_array directly

IntelPython · Sep 17, 2024 · b8dad1e · b8dad1e
1 parent 39ac075
commit b8dad1e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dpnp/dpnp_algo/dpnp_fill.py b/dpnp/dpnp_algo/dpnp_fill.py
@@ -34,6 +34,7 @@
 from dpctl.utils import SequentialOrderManager
 
 import dpnp
+from dpnp.dpnp_array import dpnp_array
 
 
 def dpnp_fill(arr, val):
@@ -43,7 +44,7 @@ def dpnp_fill(arr, val):
 
     dpnp.check_supported_arrays_type(val, scalar_type=True, all_scalars=True)
     # if val is an array, process it
-    if isinstance(val, (dpnp.dpnp_array, dpt.usm_ndarray)):
+    if isinstance(val, (dpnp_array, dpt.usm_ndarray)):
         val = dpnp.get_usm_ndarray(val)
         if val.shape != ():
             raise ValueError("`val` must be a scalar")