MAINT: following #1936 for sklearnex/_device_offload.py (#2137)

uxlfoundation · Nov 4, 2024 · e473fd5 · e473fd5
1 parent 08656de
commit e473fd5
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/sklearnex/_device_offload.py b/sklearnex/_device_offload.py
@@ -16,13 +16,9 @@
 
 from functools import wraps
 
-from onedal._device_offload import (
-    _copy_to_usm,
-    _get_global_queue,
-    _transfer_to_host,
-    dpnp_available,
-)
+from onedal._device_offload import _copy_to_usm, _get_global_queue, _transfer_to_host
 from onedal.utils._array_api import _asarray
+from onedal.utils._dpep_helpers import dpnp_available
 
 if dpnp_available:
     import dpnp