oscura 0.1.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

oscura/core/numba_backend.py

@@ -35,8 +35,6 @@ Example:
     >>> result = sum_of_squares(data)  # Fast on second call
 """
 
-from __future__ import annotations
-
 import functools
 from collections.abc import Callable
 from typing import Any, TypeVar
@@ -45,11 +43,11 @@ import numpy as np
 
 # Try to import Numba
 try:
-    from numba import guvectorize as _numba_guvectorize  # type: ignore[import-untyped]
-    from numba import jit as _numba_jit  # type: ignore[import-untyped]
-    from numba import njit as _numba_njit  # type: ignore[import-untyped]
-    from numba import prange as _numba_prange  # type: ignore[import-untyped]
-    from numba import vectorize as _numba_vectorize  # type: ignore[import-untyped]
+    from numba import guvectorize as _numba_guvectorize  # type: ignore[import-not-found]
+    from numba import jit as _numba_jit  # type: ignore[import-not-found]
+    from numba import njit as _numba_njit  # type: ignore[import-not-found]
+    from numba import prange as _numba_prange  # type: ignore[import-not-found]
+    from numba import vectorize as _numba_vectorize  # type: ignore[import-not-found]
 
     HAS_NUMBA = True
 except ImportError:

oscura/core/uncertainty.py

@@ -108,7 +108,7 @@ class MeasurementWithUncertainty:
             JCGM 100:2008 Section 5.1.6
         """
         if self.value == 0:
-            return np.inf
+            return float(np.inf)  # type: ignore[no-any-return]
         return abs(self.uncertainty / self.value)
 
     @property
@@ -173,7 +173,7 @@ class UncertaintyEstimator:
             JCGM 100:2008 Section 4.2 (Type A evaluation)
         """
         if len(data) < 2:
-            return np.nan
+            return float(np.nan)  # type: ignore[no-any-return]
         return float(np.std(data, ddof=1))  # Sample std (Bessel correction)
 
     @staticmethod
@@ -190,7 +190,7 @@ class UncertaintyEstimator:
             JCGM 100:2008 Section 4.2.3
         """
         if len(data) < 2:
-            return np.nan
+            return float(np.nan)  # type: ignore[no-any-return]
         return float(np.std(data, ddof=1) / np.sqrt(len(data)))
 
     @staticmethod

oscura/dsl/interpreter.py

@@ -3,6 +3,8 @@
 Executes parsed DSL programs.
 """
 
+from __future__ import annotations
+
 from collections.abc import Callable
 from pathlib import Path
 from typing import Any

oscura/dsl/parser.py

@@ -3,6 +3,8 @@
 Implements simple domain-specific language for trace analysis workflows.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from enum import Enum, auto
 from typing import Any, Union
@@ -352,14 +354,14 @@ class Assignment(ASTNode):
     """Variable assignment: $var = expr."""
 
     variable: str
-    expression: "Expression"
+    expression: Expression
 
 
 @dataclass
 class Pipeline(ASTNode):
     """Pipeline expression: expr | command | command."""
 
-    stages: list["Expression"]
+    stages: list[Expression]
 
 
 @dataclass
@@ -367,7 +369,7 @@ class Command(ASTNode):
     """Command invocation: command arg1 arg2."""
 
     name: str
-    args: list["Expression"]
+    args: list[Expression]
 
 
 @dataclass
@@ -375,7 +377,7 @@ class FunctionCall(ASTNode):
     """Function call: func(arg1, arg2)."""
 
     name: str
-    args: list["Expression"]
+    args: list[Expression]
 
 
 @dataclass
@@ -397,8 +399,8 @@ class ForLoop(ASTNode):
     """For loop: for $var in expr: body."""
 
     variable: str
-    iterable: "Expression"
-    body: list["Statement"]
+    iterable: Expression
+    body: list[Statement]
 
 
 # Type aliases

oscura/exploratory/error_recovery.py

@@ -20,6 +20,8 @@ import numpy as np
 
 from oscura.core.types import WaveformTrace
 
+T = TypeVar("T")
+
 logger = logging.getLogger(__name__)
 
 if TYPE_CHECKING:
@@ -27,8 +29,6 @@ if TYPE_CHECKING:
 
     from numpy.typing import NDArray
 
-T = TypeVar("T")
-
 
 @dataclass
 class RecoveryStats:
@@ -547,7 +547,7 @@ class RetryResult:
     adjustments_made: list[str]
 
 
-def retry_with_adjustment[T](
+def retry_with_adjustment(
     func: Callable[..., T],
     trace: WaveformTrace,
     initial_params: dict[str, Any],

oscura/exploratory/parse.py

@@ -5,6 +5,8 @@ This module provides robust protocol decoding that continues after errors
 and timestamp correction for jittery captures.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from enum import Enum
 from typing import Any, Literal

oscura/exploratory/recovery.py

@@ -5,6 +5,8 @@ This module characterizes bit error patterns to diagnose capture quality
 issues (EMI, USB problems, clock jitter) and suggests likely causes.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from enum import Enum
 

oscura/exploratory/sync.py

@@ -6,6 +6,8 @@ in noisy or corrupted logic analyzer captures, with configurable bit error
 tolerance using Hamming distance.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from enum import Enum
 from typing import Literal

oscura/export/wireshark/generator.py

@@ -106,7 +106,7 @@ class WiresharkDissectorGenerator:
 
         # Render template
         template = self.env.get_template("dissector.lua.j2")
-        return template.render(**context)
+        return str(template.render(**context))  # type: ignore[no-any-return]
 
     def _build_template_context(self, protocol: ProtocolDefinition) -> dict[str, Any]:
         """Build context dictionary for template rendering.

oscura/export/wireshark/type_mapping.py

@@ -7,6 +7,8 @@ References:
     https://wiki.wireshark.org/LuaAPI/Proto
 """
 
+from __future__ import annotations
+
 from typing import Literal
 
 # Map Oscura field types to Wireshark ProtoField types

oscura/exporters/hdf5.py

@@ -11,8 +11,6 @@ References:
     HDF5 specification (https://www.hdfgroup.org/)
 """
 
-from __future__ import annotations
-
 from datetime import datetime
 from pathlib import Path
 from typing import Any
@@ -81,7 +79,7 @@ def export_hdf5(
 
 
 def _write_trace_dataset(
-    f: h5py.File,
+    f: "h5py.File",
     name: str,
     trace: WaveformTrace | DigitalTrace,
     compression: str | None,

oscura/extensibility/templates.py

@@ -622,8 +622,6 @@ def _generate_analyzer_stub(template: PluginTemplate, class_name: str) -> str:
         References:
             PLUG-008: Plugin Template Generator
         """
-        from __future__ import annotations
-
         import numpy as np
         from numpy.typing import NDArray
 
@@ -697,8 +695,6 @@ def _generate_loader_stub(template: PluginTemplate, class_name: str) -> str:
         References:
             PLUG-008: Plugin Template Generator
         """
-        from __future__ import annotations
-
         import numpy as np
         from numpy.typing import NDArray
         from pathlib import Path
@@ -782,8 +778,6 @@ def _generate_exporter_stub(template: PluginTemplate, class_name: str) -> str:
         References:
             PLUG-008: Plugin Template Generator
         """
-        from __future__ import annotations
-
         import numpy as np
         from numpy.typing import NDArray
         from pathlib import Path
@@ -871,8 +865,6 @@ def _generate_generic_stub(template: PluginTemplate, class_name: str) -> str:
         References:
             PLUG-008: Plugin Template Generator
         """
-        from __future__ import annotations
-
 
         class {class_name}:
             """Generic plugin implementation.

oscura/inference/active_learning/lstar.py

@@ -26,11 +26,9 @@ Complexity:
     - Produces minimal DFA (fewest states)
 """
 
-from __future__ import annotations
-
 from oscura.inference.active_learning.observation_table import ObservationTable
-from oscura.inference.active_learning.oracle import Oracle  # noqa: TC001
-from oscura.inference.state_machine import FiniteAutomaton  # noqa: TC001
+from oscura.inference.active_learning.oracle import Oracle
+from oscura.inference.state_machine import FiniteAutomaton
 
 
 class LStarLearner:

oscura/inference/active_learning/observation_table.py

@@ -8,8 +8,6 @@ References:
     Information and Computation, 75(2), 87-106.
 """
 
-from __future__ import annotations
-
 from dataclasses import dataclass, field
 
 from oscura.inference.state_machine import FiniteAutomaton, State, Transition

oscura/inference/active_learning/oracle.py

@@ -11,8 +11,10 @@ References:
 from __future__ import annotations
 
 from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
 
-from oscura.inference.state_machine import FiniteAutomaton  # noqa: TC001
+if TYPE_CHECKING:
+    from oscura.inference.state_machine import FiniteAutomaton
 
 
 class Oracle(ABC):

oscura/inference/active_learning/teachers/simulator.py

@@ -7,10 +7,8 @@ requiring a live system.
 The simulator teacher treats traces as examples of valid protocol sequences.
 """
 
-from __future__ import annotations
-
 from oscura.inference.active_learning.oracle import Oracle
-from oscura.inference.state_machine import FiniteAutomaton  # noqa: TC001
+from oscura.inference.state_machine import FiniteAutomaton
 
 
 class SimulatorTeacher(Oracle):

oscura/inference/alignment.py

@@ -12,6 +12,8 @@ Key capabilities:
 - Conserved/variable region detection
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from typing import Any, Literal
 

oscura/inference/message_format.py

@@ -18,6 +18,8 @@ References:
     Discoverer: Automatic Protocol Reverse Engineering. USENIX Security 2007.
 """
 
+from __future__ import annotations
+
 from dataclasses import dataclass
 from dataclasses import field as dataclass_field
 from typing import Any, Literal

oscura/inference/protocol_dsl.py

@@ -13,6 +13,8 @@ Key capabilities:
 - Comprehensive error reporting
 """
 
+from __future__ import annotations
+
 import ast
 import operator
 import struct
@@ -68,7 +70,7 @@ class FieldDefinition:
     element: dict[str, Any] | None = None  # Element definition for arrays
     count_field: str | None = None  # Field containing array count
     count: int | None = None  # Fixed array count
-    fields: list["FieldDefinition"] | None = None  # Nested fields for struct type
+    fields: list[FieldDefinition] | None = None  # Nested fields for struct type
 
     def __post_init__(self) -> None:
         """Handle size_ref as alias for size."""
@@ -117,7 +119,7 @@ class ProtocolDefinition:
     encoding: dict[str, Any] = field(default_factory=dict)
 
     @classmethod
-    def from_yaml(cls, path: str | Path) -> "ProtocolDefinition":
+    def from_yaml(cls, path: str | Path) -> ProtocolDefinition:
         """Load protocol definition from YAML file.
 
         : YAML parsing.
@@ -134,7 +136,7 @@ class ProtocolDefinition:
         return cls.from_dict(config)
 
     @classmethod
-    def from_dict(cls, config: dict[str, Any]) -> "ProtocolDefinition":
+    def from_dict(cls, config: dict[str, Any]) -> ProtocolDefinition:
         """Create from dictionary.
 
         : Configuration parsing.
@@ -168,7 +170,7 @@ class ProtocolDefinition:
     @classmethod
     def _parse_field_definition(
         cls, field_dict: dict[str, Any], default_endian: str
-    ) -> "FieldDefinition":
+    ) -> FieldDefinition:
         """Parse a single field definition from dictionary.
 
         Args:
@@ -365,7 +367,7 @@ class ProtocolDecoder:
         self._endian_map: dict[str, str] = {"big": ">", "little": "<"}
 
     @classmethod
-    def load(cls, path: str | Path) -> "ProtocolDecoder":
+    def load(cls, path: str | Path) -> ProtocolDecoder:
         """Load decoder from YAML protocol definition.
 
         : Load decoder from file.

oscura/inference/sequences.py

@@ -8,8 +8,6 @@ streams, identifying request-response pairs, and analyzing communication
 flows.
 """
 
-from __future__ import annotations
-
 from collections import defaultdict
 from collections.abc import Callable, Sequence
 from dataclasses import dataclass, field
@@ -83,7 +81,7 @@ class CommunicationFlow:
 
     flow_id: int
     messages: list[Any]
-    pairs: list[RequestResponsePair]
+    pairs: list["RequestResponsePair"]
     direction: str
     participants: list[str]
     duration: float
@@ -127,7 +125,7 @@ class SequencePatternDetector:
         messages: Sequence[Any],
         key: Callable[[Any], Any] | None = None,
         timestamp_key: Callable[[Any], float] | None = None,
-    ) -> list[SequencePattern]:
+    ) -> list["SequencePattern"]:
         """Detect sequential patterns in message stream.
 
         Implements RE-SEQ-002: Pattern detection workflow.
@@ -222,7 +220,7 @@ class SequencePatternDetector:
         messages: Sequence[Any],
         key: Callable[[Any], Any] | None = None,
         timestamp_key: Callable[[Any], float] | None = None,
-    ) -> list[SequencePattern]:
+    ) -> list["SequencePattern"]:
         """Detect patterns that occur at regular intervals.
 
         Implements RE-SEQ-002: Periodic pattern detection.
@@ -287,7 +285,7 @@ class SequencePatternDetector:
         candidates: dict[tuple[Any, ...], list[int]],
         identifiers: list[Any],
         timestamps: list[float] | None,
-    ) -> list[SequencePattern]:
+    ) -> list["SequencePattern"]:
         """Score candidate patterns.
 
         Args:
@@ -386,7 +384,7 @@ class RequestResponseCorrelator:
         request_filter: Callable[[Any], bool] | None = None,
         response_filter: Callable[[Any], bool] | None = None,
         timestamp_key: Callable[[Any], float] | None = None,
-    ) -> list[RequestResponsePair]:
+    ) -> list["RequestResponsePair"]:
         """Correlate requests with responses.
 
         Implements RE-SEQ-003: Request-response correlation workflow.
@@ -441,7 +439,7 @@ class RequestResponseCorrelator:
         messages: Sequence[Any],
         content_key: Callable[[Any], bytes],
         timestamp_key: Callable[[Any], float] | None = None,
-    ) -> list[RequestResponsePair]:
+    ) -> list["RequestResponsePair"]:
         """Correlate by analyzing message content similarity.
 
         Implements RE-SEQ-003: Content-based correlation.
@@ -500,10 +498,10 @@ class RequestResponseCorrelator:
 
     def extract_flows(
         self,
-        pairs: Sequence[RequestResponsePair],
+        pairs: Sequence["RequestResponsePair"],
         messages: Sequence[Any],
         flow_key: Callable[[Any], str] | None = None,
-    ) -> list[CommunicationFlow]:
+    ) -> list["CommunicationFlow"]:
         """Extract communication flows from pairs.
 
         Implements RE-SEQ-003: Flow extraction.
@@ -569,7 +567,7 @@ class RequestResponseCorrelator:
         self,
         requests: list[tuple[int, Any, float, Any]],
         responses: list[tuple[int, Any, float, Any]],
-    ) -> list[RequestResponsePair]:
+    ) -> list["RequestResponsePair"]:
         """Match request and response messages.
 
         Args:
@@ -664,7 +662,7 @@ def detect_sequence_patterns(
     min_length: int = 2,
     max_length: int = 10,
     min_frequency: int = 2,
-) -> list[SequencePattern]:
+) -> list["SequencePattern"]:
     """Detect sequential patterns in messages.
 
     Implements RE-SEQ-002: Sequence Pattern Detection.
@@ -699,7 +697,7 @@ def correlate_requests(
     response_filter: Callable[[Any], bool],
     timestamp_key: Callable[[Any], float] | None = None,
     max_latency: float = 10.0,
-) -> list[RequestResponsePair]:
+) -> list["RequestResponsePair"]:
     """Correlate request and response messages.
 
     Implements RE-SEQ-003: Request-Response Correlation.
@@ -760,7 +758,7 @@ def find_message_dependencies(
 
 
 def calculate_latency_stats(
-    pairs: Sequence[RequestResponsePair],
+    pairs: Sequence["RequestResponsePair"],
 ) -> dict[str, float]:
     """Calculate latency statistics for request-response pairs.
 

oscura/inference/state_machine.py

@@ -12,6 +12,8 @@ Key capabilities:
 - Export to NetworkX graph for analysis
 """
 
+from __future__ import annotations
+
 from copy import deepcopy
 from dataclasses import dataclass
 from typing import Any

oscura/integrations/llm.py

@@ -23,6 +23,8 @@ Examples:
     ... )
 """
 
+from __future__ import annotations
+
 import hashlib
 import json
 import os
@@ -1545,7 +1547,7 @@ def get_client_auto(**config_kwargs: Any) -> LLMClient:
 
 def get_client_with_failover(
     providers: list[str] | None = None, **config_kwargs: Any
-) -> "FailoverLLMClient":
+) -> FailoverLLMClient:
     """Get LLM client with automatic failover between providers.
 
     Failover logic (try OpenAI, fallback to Anthropic).

oscura/jupyter/display.py

@@ -12,8 +12,6 @@ Example:
     In [2]: display_trace(trace)  # Shows rich HTML summary
 """
 
-from __future__ import annotations
-
 from typing import Any
 
 try:

oscura/loaders/__init__.py

@@ -20,10 +20,74 @@ from __future__ import annotations
 import logging
 import warnings
 from pathlib import Path
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, cast
 
 from oscura.core.exceptions import LoaderError, UnsupportedFormatError
-from oscura.core.types import DigitalTrace, WaveformTrace
+from oscura.core.types import DigitalTrace, IQTrace, WaveformTrace
+
+# Loader registry for cleaner dispatch
+_LOADER_REGISTRY: dict[str, tuple[str, str]] = {
+    "tektronix": ("oscura.loaders.tektronix", "load_tektronix_wfm"),
+    "tek": ("oscura.loaders.tektronix", "load_tektronix_wfm"),
+    "rigol": ("oscura.loaders.rigol", "load_rigol_wfm"),
+    "numpy": ("oscura.loaders.numpy_loader", "load_npz"),
+    "csv": ("oscura.loaders.csv_loader", "load_csv"),
+    "hdf5": ("oscura.loaders.hdf5_loader", "load_hdf5"),
+    "sigrok": ("oscura.loaders.sigrok", "load_sigrok"),
+    "vcd": ("oscura.loaders.vcd", "load_vcd"),
+    "pcap": ("oscura.loaders.pcap", "load_pcap"),
+    "wav": ("oscura.loaders.wav", "load_wav"),
+    "tdms": ("oscura.loaders.tdms", "load_tdms"),
+    "touchstone": ("oscura.loaders.touchstone", "load_touchstone"),
+    "chipwhisperer": ("oscura.loaders.chipwhisperer", "load_chipwhisperer"),
+}
+
+
+def _dispatch_loader(
+    loader_name: str, path: Path, **kwargs: Any
+) -> WaveformTrace | DigitalTrace | IQTrace:
+    """Dispatch to registered loader.
+
+    Args:
+        loader_name: Name of loader to use.
+        path: Path to file.
+        **kwargs: Additional arguments for loader.
+
+    Returns:
+        Loaded data.
+
+    Raises:
+        UnsupportedFormatError: If loader not registered.
+    """
+    if loader_name not in _LOADER_REGISTRY:
+        raise UnsupportedFormatError(
+            loader_name,
+            list(_LOADER_REGISTRY.keys()),
+            file_path=str(path),
+        )
+
+    module_path, func_name = _LOADER_REGISTRY[loader_name]
+
+    # Dynamically import the module
+    import importlib
+    import inspect
+
+    module = importlib.import_module(module_path)
+    loader_func = getattr(module, func_name)
+
+    # Filter kwargs to only include parameters the function accepts
+    sig = inspect.signature(loader_func)
+    valid_kwargs = {}
+    for key, value in kwargs.items():
+        if key in sig.parameters or any(
+            p.kind == inspect.Parameter.VAR_KEYWORD for p in sig.parameters.values()
+        ):
+            valid_kwargs[key] = value
+
+    # Call loader with appropriate arguments
+    result = loader_func(path, **valid_kwargs)
+    return cast("WaveformTrace | DigitalTrace | IQTrace", result)
+
 
 # Import alias modules for DSL compatibility
 from oscura.loaders import (
@@ -180,56 +244,9 @@ def load(
     # Dispatch to appropriate loader
     if loader_name == "auto_wfm":
         return _load_wfm_auto(path, channel=channel, **kwargs)
-    elif loader_name in ("tektronix", "tek"):
-        from oscura.loaders.tektronix import load_tektronix_wfm
-
-        return load_tektronix_wfm(path, **kwargs)
-    elif loader_name == "rigol":
-        from oscura.loaders.rigol import load_rigol_wfm
-
-        return load_rigol_wfm(path, **kwargs)
-    elif loader_name == "numpy":
-        from oscura.loaders.numpy_loader import load_npz
-
-        return load_npz(path, channel=channel, **kwargs)
-    elif loader_name == "csv":
-        from oscura.loaders.csv_loader import load_csv
-
-        return load_csv(path, **kwargs)  # type: ignore[return-value]
-    elif loader_name == "hdf5":
-        from oscura.loaders.hdf5_loader import load_hdf5
-
-        return load_hdf5(path, channel=channel, **kwargs)  # type: ignore[return-value]
-    elif loader_name == "sigrok":
-        from oscura.loaders.sigrok import load_sigrok
-
-        return load_sigrok(path, channel=channel, **kwargs)
-    elif loader_name == "vcd":
-        from oscura.loaders.vcd import load_vcd
-
-        return load_vcd(path, **kwargs)
-    elif loader_name == "pcap":
-        from oscura.loaders.pcap import load_pcap
-
-        return load_pcap(path, **kwargs)  # type: ignore[return-value]
-    elif loader_name == "wav":
-        from oscura.loaders.wav import load_wav
-
-        return load_wav(path, channel=channel, **kwargs)
-    elif loader_name == "tdms":
-        from oscura.loaders.tdms import load_tdms
-
-        return load_tdms(path, channel=channel, **kwargs)
-    elif loader_name == "touchstone":
-        from oscura.analyzers.signal_integrity.sparams import load_touchstone
-
-        return load_touchstone(path)  # type: ignore[return-value]
     else:
-        raise UnsupportedFormatError(
-            loader_name,
-            list(SUPPORTED_FORMATS.keys()),
-            file_path=str(path),
-        )
+        # Use registry-based dispatch for all other loaders
+        return _dispatch_loader(loader_name, path, channel=channel, **kwargs)
 
 
 def _load_wfm_auto(