Address review comments.

Author: Diptorup Deb

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
 

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