Update import

IntelPython · Mar 8, 2024 · c3e02d7 · c3e02d7
1 parent 7f75500
commit c3e02d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numba_dpex/dpctl_iface/_helpers.py b/numba_dpex/dpctl_iface/_helpers.py
@@ -14,7 +14,7 @@ def numba_type_to_dpctl_typenum(context, ty):
     """
 
     if dpctl_sem_version >= (0, 17, 0):
-        from dpctl.enum_types import kernel_arg_type as kargty
+        from dpctl._sycl_queue import kernel_arg_type as kargty
 
         if ty == types.int32 or isinstance(ty, types.scalars.IntegerLiteral):
             return context.get_constant(types.int32, kargty.dpctl_int32.value)