Address review comments.

IntelPython · Dec 7, 2023 · 9ffb10e · 9ffb10e
1 parent fc06d52
commit 9ffb10e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/numba_dpex/experimental/models.py b/numba_dpex/experimental/models.py
@@ -17,7 +17,7 @@
 from .types import KernelDispatcherType
 
 
-class LiteralIntEnumModel(PrimitiveModel):
+class IntEnumLiteralModel(PrimitiveModel):
     """Representation of an object of LiteralIntEnum type using Numba's
     PrimitiveModel that can be represented natively in the target in all
     usage contexts.
@@ -42,7 +42,7 @@ def _init_exp_data_model_manager() -> DataModelManager:
     dmm = dpex_core_models.dpex_data_model_manager.copy()
 
     # Register the types and data model in the DpexExpTargetContext
-    dmm.register(IntEnumLiteral, LiteralIntEnumModel)
+    dmm.register(IntEnumLiteral, IntEnumLiteralModel)
 
     return dmm
 

diff --git a/numba_dpex/tests/experimental/IntEnumLiteral/test_compilation.py b/numba_dpex/tests/experimental/IntEnumLiteral/test_compilation.py
@@ -3,11 +3,9 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import dpnp
-from numba.core import types
-from numba.extending import intrinsic, overload
 
 import numba_dpex.experimental as exp_dpex
-from numba_dpex import Range, dpjit
+from numba_dpex import Range
 from numba_dpex.experimental.flag_enum import FlagEnum