numba_type_utils -> numba_cfunc_compiler

Author: Halka

conda/dev-environment-unix.yml

@@ -13,7 +13,7 @@ dependencies:
  - cyrus-sasl
  - deprecated
  - docutils<0.22.1
- - numba_type_utils
+ - numba_cfunc_compiler
  - exprtk
  - flex
  - graphviz

conda/dev-environment-win.yml

@@ -13,7 +13,7 @@ dependencies:
  - cyrus-sasl
  - deprecated
  - docutils<0.22.1
- - numba_type_utils
+ - numba_cfunc_compiler
  - exprtk
 #  - flex  # not available on windows
  - graphviz

csp/impl/wiring/csp_numba/enum_support.py

@@ -5,10 +5,10 @@
 
 from csp.impl.enum import Enum as CspEnum
 
-from numba_type_utils.type_factory import TypeFactory
-from numba_type_utils.models import VariableType, ParameterInfo, StateVariableInfo
-from numba_type_utils.numba_type_inference import NumbaTypeInference
-from numba_type_utils.utils.ast import AST
+from numba_cfunc_compiler.type_factory import TypeFactory
+from numba_cfunc_compiler.models import VariableType, ParameterInfo, StateVariableInfo
+from numba_cfunc_compiler.numba_type_inference import NumbaTypeInference
+from numba_cfunc_compiler.utils.ast import AST
 
 
 def _resolve_csp_enum_member(node: ast.AST, globalns: dict) -> Optional[CspEnum]:

csp/impl/wiring/csp_numba/input_handlers.py

@@ -13,8 +13,8 @@
 from csp.impl.wiring.edge import Edge
 from csp.impl.types.tstype import TsType
 
-from numba_type_utils.function_analyzer import InputTypeHandler
-from numba_type_utils.models import ParameterInfo
+from numba_cfunc_compiler.function_analyzer import InputTypeHandler
+from numba_cfunc_compiler.models import ParameterInfo
 
 
 def _extract_ts_inner_type(ann: Any) -> Optional[type]:
@@ -130,7 +130,7 @@ def validate_value(self, param_name: str, value: Any, expected_type: Any) -> Any
 
 
 def register():
-    from numba_type_utils.function_analyzer import FunctionAnalyzer
+    from numba_cfunc_compiler.function_analyzer import FunctionAnalyzer
 
     FunctionAnalyzer.register_input_handler(TsInputHandler())
     FunctionAnalyzer.register_input_handler(ListBasketInputHandler())

csp/impl/wiring/csp_numba/numba_node.py

@@ -22,12 +22,12 @@
     register_ast_handlers as register_signal_set_ast_handlers,
 )
 
-from numba_type_utils.compilation_context import CompilationContext
-from numba_type_utils.numba_core import create_compiled_func
-from numba_type_utils.numba_config import State, set_output
-from numba_type_utils.defaults import register_types as register_default_types
-from numba_type_utils.source_registry import SourceRegistry
-from numba_type_utils.defaults.source_categories import (
+from numba_cfunc_compiler.compilation_context import CompilationContext
+from numba_cfunc_compiler.numba_core import create_compiled_func
+from numba_cfunc_compiler.numba_config import State, set_output
+from numba_cfunc_compiler.defaults import register_types as register_default_types
+from numba_cfunc_compiler.source_registry import SourceRegistry
+from numba_cfunc_compiler.defaults.source_categories import (
     ConstantCategory,
     OutputCategory,
     StateCategory,

csp/impl/wiring/csp_numba/output_handlers.py

@@ -12,9 +12,9 @@
 from csp.impl.struct import Struct as CspStruct
 from csp.impl.types.tstype import TsType
 
-from numba_type_utils.function_analyzer import OutputTypeHandler
-from numba_type_utils.models import OutputAnalysis
-from numba_type_utils.defaults import is_supported_type
+from numba_cfunc_compiler.function_analyzer import OutputTypeHandler
+from numba_cfunc_compiler.models import OutputAnalysis
+from numba_cfunc_compiler.defaults import is_supported_type
 
 
 def validate_output_type(t: Any, context: str) -> None:
@@ -82,7 +82,7 @@ def try_parse(self, return_annotation: Any, ast_tree: ast.AST) -> Optional[Outpu
 
 
 def register():
-    from numba_type_utils.function_analyzer import FunctionAnalyzer
+    from numba_cfunc_compiler.function_analyzer import FunctionAnalyzer
 
     FunctionAnalyzer.register_output_handler(SingleTsOutputHandler())
     FunctionAnalyzer.register_output_handler(CspOutputsHandler())

csp/impl/wiring/csp_numba/signal_set_support.py

@@ -1,10 +1,10 @@
 import ast
 from typing import Optional
 
-from numba_type_utils.utils.ast import AST
-from numba_type_utils.method_factory import MethodBase
-from numba_type_utils.variable_factory import VariableSource
-from numba_type_utils.models import UnknownType, UnknownNumbaType
+from numba_cfunc_compiler.utils.ast import AST
+from numba_cfunc_compiler.method_factory import MethodBase
+from numba_cfunc_compiler.variable_factory import VariableSource
+from numba_cfunc_compiler.models import UnknownType, UnknownNumbaType
 
 from csp.impl.wiring.csp_numba.signal_support import (
     INPUTS_ARRAY_NAME,
@@ -401,7 +401,7 @@ def handle_signal_set_iteration(converter, node: ast.For) -> Optional[ast.AST]:
 
 
 def register_ast_handlers():
-    from numba_type_utils.ast_handlers import ASTHandlerRegistry, HandlerPhase
+    from numba_cfunc_compiler.ast_handlers import ASTHandlerRegistry, HandlerPhase
 
     ASTHandlerRegistry.register("Call", handle_len_signal_set, HandlerPhase.PRE, priority=0)
     ASTHandlerRegistry.register("Subscript", handle_signal_set_subscript, HandlerPhase.PRE, priority=0)

csp/impl/wiring/csp_numba/signal_support.py

@@ -1,9 +1,9 @@
 import ast
 
-from numba_type_utils.utils.ast import AST
-from numba_type_utils.method_factory import MethodBase
-from numba_type_utils.variable_factory import VoidPtrSource
-from numba_type_utils.source_registry import CfuncParam, SourceCategory, SourceInitFilter
+from numba_cfunc_compiler.utils.ast import AST
+from numba_cfunc_compiler.method_factory import MethodBase
+from numba_cfunc_compiler.variable_factory import VoidPtrSource
+from numba_cfunc_compiler.source_registry import CfuncParam, SourceCategory, SourceInitFilter
 
 __all__ = [
     "INPUTS_ARRAY_NAME",
@@ -144,7 +144,7 @@ def _create_signal_source(
         )
 
     def create_variables(self, info, factory):
-        from numba_type_utils.type_factory import TypeFactory
+        from numba_cfunc_compiler.type_factory import TypeFactory
         from csp.impl.wiring.csp_numba.signal_set_support import SignalSetSource
 
         input_idx = 0

csp/impl/wiring/csp_numba/struct_support.py

@@ -8,13 +8,13 @@
 
 from csp.impl.struct import Struct as CspStruct
 
-from numba_type_utils.type_factory import TypeFactory
-from numba_type_utils.models import StateVariableInfo
-from numba_type_utils.defaults.struct_support import StructType, StructFieldInfo
-from numba_type_utils.numba_config import NumbaTypeRegistry
-from numba_type_utils.numba_type_inference import NumbaTypeInference
-from numba_type_utils.defaults.struct_support import struct_attribute_transformer, struct_attr_handler
-from numba_type_utils.utils.ast import AST
+from numba_cfunc_compiler.type_factory import TypeFactory
+from numba_cfunc_compiler.models import StateVariableInfo
+from numba_cfunc_compiler.defaults.struct_support import StructType, StructFieldInfo
+from numba_cfunc_compiler.numba_config import NumbaTypeRegistry
+from numba_cfunc_compiler.numba_type_inference import NumbaTypeInference
+from numba_cfunc_compiler.defaults.struct_support import struct_attribute_transformer, struct_attr_handler
+from numba_cfunc_compiler.utils.ast import AST
 
 
 def _is_csp_struct_subclass(value: Any) -> bool:

csp/tests/test_numba_node.py

@@ -4,7 +4,7 @@
 import csp
 from csp import ts
 from csp.impl.wiring.csp_numba import numba_node
-from numba_type_utils.numba_config import NumbaList, NumbaDict, create_new_list, create_new_dict
+from numba_cfunc_compiler.numba_config import NumbaList, NumbaDict, create_new_list, create_new_dict
 
 
 class TestBasicTypes(unittest.TestCase):