fix lint

Signed-off-by: Mark Halka <mark.halka2001@gmail.com>

Author: markhalka

csp/impl/wiring/__init__.py

@@ -1,13 +1,13 @@
 from csp.impl.wiring.adapters import *
 from csp.impl.wiring.context import *
+from csp.impl.wiring.csp_numba.numba_node import *
 from csp.impl.wiring.delayed_edge import DelayedEdge
 from csp.impl.wiring.dynamic import dynamic
 from csp.impl.wiring.edge import *
 from csp.impl.wiring.feedback import *
 from csp.impl.wiring.graph import *
 from csp.impl.wiring.node import *
 from csp.impl.wiring.node_parser import *
-from csp.impl.wiring.csp_numba.numba_node import *
 from csp.impl.wiring.outputs import *
 from csp.impl.wiring.runtime import *
 from csp.impl.wiring.signature import *

csp/impl/wiring/csp_numba/__init__.py

@@ -1,5 +1,5 @@
-from .csp_node_transformer import CspNodeTransformer, TransformedNode, StateVariable
-from .numba_node import numba_node, NumbaNodeDef
+from .csp_node_transformer import CspNodeTransformer, StateVariable, TransformedNode
+from .numba_node import NumbaNodeDef, numba_node
 
 __all__ = [
     "numba_node",

csp/impl/wiring/csp_numba/enum_support.py

@@ -3,13 +3,13 @@
 from dataclasses import dataclass
 from typing import Any, Optional
 
-from csp.impl.enum import Enum as CspEnum
-
-from numba_cfunc_compiler.type_factory import TypeFactory
-from numba_cfunc_compiler.models import VariableType, ParameterInfo, StateVariableInfo
+from numba_cfunc_compiler.models import ParameterInfo, StateVariableInfo, VariableType
 from numba_cfunc_compiler.numba_type_inference import NumbaTypeInference
+from numba_cfunc_compiler.type_factory import TypeFactory
 from numba_cfunc_compiler.utils.ast import AST
 
+from csp.impl.enum import Enum as CspEnum
+
 
 def _resolve_csp_enum_member(node: ast.AST, globalns: dict) -> Optional[CspEnum]:
     """Resolve `MyEnum.VALUE` and `MyEnum.VALUE.value` AST nodes."""

csp/impl/wiring/csp_numba/input_handlers.py

@@ -10,12 +10,12 @@
 import inspect
 from typing import Any, Optional, get_args, get_origin
 
-from csp.impl.wiring.edge import Edge
-from csp.impl.types.tstype import TsType
-
 from numba_cfunc_compiler.function_analyzer import InputTypeHandler
 from numba_cfunc_compiler.models import ParameterInfo
 
+from csp.impl.types.tstype import TsType
+from csp.impl.wiring.edge import Edge
+
 
 def _extract_ts_inner_type(ann: Any) -> Optional[type]:
     if get_origin(ann) is not TsType:

csp/impl/wiring/csp_numba/numba_node.py

@@ -3,36 +3,34 @@
 import inspect
 from typing import Callable, List, Optional, Tuple
 
+from numba_cfunc_compiler.compilation_context import CompilationContext
+from numba_cfunc_compiler.defaults import register_types as register_default_types
+from numba_cfunc_compiler.defaults.source_categories import (
+    ConstantCategory,
+    LifecycleCategory,
+    OutputCategory,
+    StateCategory,
+)
+from numba_cfunc_compiler.numba_core import create_compiled_func
+from numba_cfunc_compiler.source_registry import SourceRegistry
+
 from csp.impl.__cspimpl import _cspimpl
-from csp.impl.wiring.edge import Edge
-from csp.impl.types.tstype import ts
 from csp.impl.types.container_type_normalizer import ContainerTypeNormalizer
-
+from csp.impl.types.tstype import ts
 from csp.impl.wiring.csp_numba.csp_node_transformer import CspNodeTransformer
+from csp.impl.wiring.csp_numba.enum_support import register as register_enum_support
 from csp.impl.wiring.csp_numba.input_handlers import register as register_input_handlers
 from csp.impl.wiring.csp_numba.output_handlers import register as register_output_handlers
-from csp.impl.wiring.csp_numba.enum_support import register as register_enum_support
+from csp.impl.wiring.csp_numba.signal_set_support import (
+    register_ast_handlers as register_signal_set_ast_handlers,
+)
+from csp.impl.wiring.csp_numba.signal_support import SignalCategory
 from csp.impl.wiring.csp_numba.struct_support import (
     register as register_struct_support,
     struct_enum_store,
     struct_enum_value,
 )
-from csp.impl.wiring.csp_numba.signal_support import SignalCategory
-from csp.impl.wiring.csp_numba.signal_set_support import (
-    register_ast_handlers as register_signal_set_ast_handlers,
-)
-
-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,
-    LifecycleCategory,
-)
+from csp.impl.wiring.edge import Edge
 
 __all__ = (
     "numba_node",

csp/impl/wiring/csp_numba/output_handlers.py

@@ -9,12 +9,12 @@
 import ast
 from typing import Any, Optional, get_args, get_origin
 
-from csp.impl.struct import Struct as CspStruct
-from csp.impl.types.tstype import TsType
-
+from numba_cfunc_compiler.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
+
+from csp.impl.struct import Struct as CspStruct
+from csp.impl.types.tstype import TsType
 
 
 def validate_output_type(t: Any, context: str) -> None:

csp/impl/wiring/csp_numba/signal_set_support.py

@@ -1,10 +1,10 @@
 import ast
 from typing import Optional
 
-from numba_cfunc_compiler.utils.ast import AST
 from numba_cfunc_compiler.method_factory import MethodBase
+from numba_cfunc_compiler.models import UnknownNumbaType, UnknownType
+from numba_cfunc_compiler.utils.ast import AST
 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,

csp/impl/wiring/csp_numba/signal_support.py

@@ -1,9 +1,9 @@
 import ast
 
-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
+from numba_cfunc_compiler.utils.ast import AST
+from numba_cfunc_compiler.variable_factory import VoidPtrSource
 
 __all__ = [
     "INPUTS_ARRAY_NAME",
@@ -145,6 +145,7 @@ def _create_signal_source(
 
     def create_variables(self, info, factory):
         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

@@ -5,17 +5,20 @@
 from llvmlite import ir
 from numba import types as numba_types
 from numba.extending import intrinsic
-
-from csp.impl.struct import Struct as CspStruct
-
-from numba_cfunc_compiler.type_factory import TypeFactory
+from numba_cfunc_compiler.defaults.struct_support import (
+    StructFieldInfo,
+    StructType,
+    struct_attr_handler,
+    struct_attribute_transformer,
+)
 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.type_factory import TypeFactory
 from numba_cfunc_compiler.utils.ast import AST
 
+from csp.impl.struct import Struct as CspStruct
+
 
 def _is_csp_struct_subclass(value: Any) -> bool:
     return isinstance(value, type) and issubclass(value, CspStruct) and value is not CspStruct
@@ -204,6 +207,7 @@ def register():
     import ctypes
 
     import llvmlite.binding as llvm
+
     from csp.impl.__cspimpl import _cspimpl
 
     TypeFactory.register(CspStructType)

csp/tests/test_numba_node.py

@@ -1,10 +1,11 @@
 import unittest
 from datetime import datetime, timedelta, timezone
 
+from numba_cfunc_compiler.numba_config import NumbaDict, NumbaList, create_new_dict, create_new_list
+
 import csp
 from csp import ts
 from csp.impl.wiring.csp_numba import numba_node
-from numba_cfunc_compiler.numba_config import NumbaList, NumbaDict, create_new_list, create_new_dict
 
 
 class TestBasicTypes(unittest.TestCase):