Skip to content

Commit e463ed6

Browse files
author
Diptorup Deb
committed
Rename helper function to clearly indicate its usage
1 parent af26784 commit e463ed6

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

numba_dpex/core/typing/typeof.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from dpctl import SyclQueue
66
from dpctl.tensor import usm_ndarray
77
from dpnp import ndarray
8-
from numba.core import types
98
from numba.extending import typeof_impl
109
from numba.np import numpy_support
1110

@@ -16,7 +15,7 @@
1615
from ..types.usm_ndarray_type import USMNdArray
1716

1817

19-
def _typeof_helper(val, array_class_type):
18+
def _array_typeof_helper(val, array_class_type):
2019
"""Creates a Numba type of the specified ``array_class_type`` for ``val``."""
2120
try:
2221
dtype = numpy_support.from_dtype(val.dtype)
@@ -74,7 +73,7 @@ def typeof_usm_ndarray(val, c):
7473
7574
Returns: The Numba type corresponding to dpctl.tensor.usm_ndarray
7675
"""
77-
return _typeof_helper(val, USMNdArray)
76+
return _array_typeof_helper(val, USMNdArray)
7877

7978

8079
@typeof_impl.register(ndarray)
@@ -92,7 +91,7 @@ def typeof_dpnp_ndarray(val, c):
9291
9392
Returns: The Numba type corresponding to dpnp.ndarray
9493
"""
95-
return _typeof_helper(val, DpnpNdArray)
94+
return _array_typeof_helper(val, DpnpNdArray)
9695

9796

9897
@typeof_impl.register(SyclQueue)

0 commit comments

Comments
 (0)