nrl-tracker 1.7.0__tar.gz → 1.7.3__tar.gz

{nrl_tracker-1.7.0/nrl_tracker.egg-info → nrl_tracker-1.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nrl-tracker
-Version: 1.7.0
+Version: 1.7.3
 Summary: Python port of the U.S. Naval Research Laboratory's Tracker Component Library for target tracking algorithms
 Author: Original: David F. Crouse, Naval Research Laboratory
 Maintainer: Python Port Contributors
@@ -63,12 +63,13 @@ Requires-Dist: nrl-tracker[astronomy,dev,geodesy,optimization,signal,visualizati
 
 # Tracker Component Library (Python)
 
-[![PyPI version](https://img.shields.io/badge/pypi-v1.7.0-blue.svg)](https://pypi.org/project/nrl-tracker/)
+[![PyPI version](https://img.shields.io/badge/pypi-v1.7.2-blue.svg)](https://pypi.org/project/nrl-tracker/)
 [![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
 [![License: Public Domain](https://img.shields.io/badge/License-Public%20Domain-brightgreen.svg)](https://en.wikipedia.org/wiki/Public_domain)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Tests](https://img.shields.io/badge/tests-1988%20passing-success.svg)](https://github.com/nedonatelli/TCL)
 [![MATLAB Parity](https://img.shields.io/badge/MATLAB%20Parity-100%25-brightgreen.svg)](docs/gap_analysis.rst)
+[![Type Checking](https://img.shields.io/badge/mypy--strict-passing-brightgreen.svg)](mypy.ini)
 
 A Python port of the [U.S. Naval Research Laboratory's Tracker Component Library](https://github.com/USNavalResearchLaboratory/TrackerComponentLibrary), a comprehensive collection of algorithms for target tracking, estimation, coordinate systems, and related mathematical functions.
 
@@ -208,7 +209,46 @@ pytcl/
 
 - [API Reference](https://pytcl.readthedocs.io/en/latest/api/)
 - [User Guides](https://pytcl.readthedocs.io/en/latest/user_guide/)
-- [Examples](examples/)
+- [Examples](examples/) - 29 validated example scripts
+- [Tutorials](docs/tutorials/) - 10 interactive tutorial modules
+
+## Examples & Tutorials
+
+The library includes 39 runnable code examples demonstrating all major features:
+
+### Examples (29 files in `/examples/`)
+
+Comprehensive demonstrations of library functionality:
+- **Tracking & Estimation**: Kalman filters, particle filters, smoothers
+- **Assignment**: Hungarian algorithm, k-best assignments, 3D assignment
+- **Coordinates**: Frame conversions, transformations, geodetic calculations
+- **Dynamics**: State models, motion models, dynamic systems
+- **Filtering**: Uncertainty visualization, multi-target tracking
+- **Astronomy**: Ephemerides, orbital mechanics, relativistic corrections
+- **Navigation**: INS/GNSS integration, geophysical modeling
+- **Signal Processing**: Detection, filtering, transforms
+- **Terrain & Atmosphere**: Elevation models, atmospheric properties
+
+**Status**: ✅ All 29 examples validated and passing (100% execution success)
+
+### Tutorials (10 modules in `/docs/tutorials/`)
+
+Interactive learning modules with visualizations:
+- Assignment algorithms and 3D assignment problems
+- Atmospheric and geophysical models
+- Dynamical systems and reference frames
+- Filtering and smoothing techniques
+- Sensor fusion and advanced filtering
+- Special functions and mathematical tools
+
+**Status**: ✅ All 10 tutorials validated and passing (100% execution success)
+
+## Documentation
+
+- [API Reference](https://pytcl.readthedocs.io/en/latest/api/)
+- [User Guides](https://pytcl.readthedocs.io/en/latest/user_guide/)
+- [Examples](examples/) - 29 validated example scripts
+- [Tutorials](docs/tutorials/) - 10 interactive tutorial modules
 
 ## Comparison with Original MATLAB Library
 

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/README.md

@@ -1,11 +1,12 @@
 # Tracker Component Library (Python)
 
-[![PyPI version](https://img.shields.io/badge/pypi-v1.7.0-blue.svg)](https://pypi.org/project/nrl-tracker/)
+[![PyPI version](https://img.shields.io/badge/pypi-v1.7.2-blue.svg)](https://pypi.org/project/nrl-tracker/)
 [![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
 [![License: Public Domain](https://img.shields.io/badge/License-Public%20Domain-brightgreen.svg)](https://en.wikipedia.org/wiki/Public_domain)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Tests](https://img.shields.io/badge/tests-1988%20passing-success.svg)](https://github.com/nedonatelli/TCL)
 [![MATLAB Parity](https://img.shields.io/badge/MATLAB%20Parity-100%25-brightgreen.svg)](docs/gap_analysis.rst)
+[![Type Checking](https://img.shields.io/badge/mypy--strict-passing-brightgreen.svg)](mypy.ini)
 
 A Python port of the [U.S. Naval Research Laboratory's Tracker Component Library](https://github.com/USNavalResearchLaboratory/TrackerComponentLibrary), a comprehensive collection of algorithms for target tracking, estimation, coordinate systems, and related mathematical functions.
 
@@ -145,7 +146,46 @@ pytcl/
 
 - [API Reference](https://pytcl.readthedocs.io/en/latest/api/)
 - [User Guides](https://pytcl.readthedocs.io/en/latest/user_guide/)
-- [Examples](examples/)
+- [Examples](examples/) - 29 validated example scripts
+- [Tutorials](docs/tutorials/) - 10 interactive tutorial modules
+
+## Examples & Tutorials
+
+The library includes 39 runnable code examples demonstrating all major features:
+
+### Examples (29 files in `/examples/`)
+
+Comprehensive demonstrations of library functionality:
+- **Tracking & Estimation**: Kalman filters, particle filters, smoothers
+- **Assignment**: Hungarian algorithm, k-best assignments, 3D assignment
+- **Coordinates**: Frame conversions, transformations, geodetic calculations
+- **Dynamics**: State models, motion models, dynamic systems
+- **Filtering**: Uncertainty visualization, multi-target tracking
+- **Astronomy**: Ephemerides, orbital mechanics, relativistic corrections
+- **Navigation**: INS/GNSS integration, geophysical modeling
+- **Signal Processing**: Detection, filtering, transforms
+- **Terrain & Atmosphere**: Elevation models, atmospheric properties
+
+**Status**: ✅ All 29 examples validated and passing (100% execution success)
+
+### Tutorials (10 modules in `/docs/tutorials/`)
+
+Interactive learning modules with visualizations:
+- Assignment algorithms and 3D assignment problems
+- Atmospheric and geophysical models
+- Dynamical systems and reference frames
+- Filtering and smoothing techniques
+- Sensor fusion and advanced filtering
+- Special functions and mathematical tools
+
+**Status**: ✅ All 10 tutorials validated and passing (100% execution success)
+
+## Documentation
+
+- [API Reference](https://pytcl.readthedocs.io/en/latest/api/)
+- [User Guides](https://pytcl.readthedocs.io/en/latest/user_guide/)
+- [Examples](examples/) - 29 validated example scripts
+- [Tutorials](docs/tutorials/) - 10 interactive tutorial modules
 
 ## Comparison with Original MATLAB Library
 

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3/nrl_tracker.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nrl-tracker
-Version: 1.7.0
+Version: 1.7.3
 Summary: Python port of the U.S. Naval Research Laboratory's Tracker Component Library for target tracking algorithms
 Author: Original: David F. Crouse, Naval Research Laboratory
 Maintainer: Python Port Contributors
@@ -63,12 +63,13 @@ Requires-Dist: nrl-tracker[astronomy,dev,geodesy,optimization,signal,visualizati
 
 # Tracker Component Library (Python)
 
-[![PyPI version](https://img.shields.io/badge/pypi-v1.7.0-blue.svg)](https://pypi.org/project/nrl-tracker/)
+[![PyPI version](https://img.shields.io/badge/pypi-v1.7.2-blue.svg)](https://pypi.org/project/nrl-tracker/)
 [![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
 [![License: Public Domain](https://img.shields.io/badge/License-Public%20Domain-brightgreen.svg)](https://en.wikipedia.org/wiki/Public_domain)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 [![Tests](https://img.shields.io/badge/tests-1988%20passing-success.svg)](https://github.com/nedonatelli/TCL)
 [![MATLAB Parity](https://img.shields.io/badge/MATLAB%20Parity-100%25-brightgreen.svg)](docs/gap_analysis.rst)
+[![Type Checking](https://img.shields.io/badge/mypy--strict-passing-brightgreen.svg)](mypy.ini)
 
 A Python port of the [U.S. Naval Research Laboratory's Tracker Component Library](https://github.com/USNavalResearchLaboratory/TrackerComponentLibrary), a comprehensive collection of algorithms for target tracking, estimation, coordinate systems, and related mathematical functions.
 
@@ -208,7 +209,46 @@ pytcl/
 
 - [API Reference](https://pytcl.readthedocs.io/en/latest/api/)
 - [User Guides](https://pytcl.readthedocs.io/en/latest/user_guide/)
-- [Examples](examples/)
+- [Examples](examples/) - 29 validated example scripts
+- [Tutorials](docs/tutorials/) - 10 interactive tutorial modules
+
+## Examples & Tutorials
+
+The library includes 39 runnable code examples demonstrating all major features:
+
+### Examples (29 files in `/examples/`)
+
+Comprehensive demonstrations of library functionality:
+- **Tracking & Estimation**: Kalman filters, particle filters, smoothers
+- **Assignment**: Hungarian algorithm, k-best assignments, 3D assignment
+- **Coordinates**: Frame conversions, transformations, geodetic calculations
+- **Dynamics**: State models, motion models, dynamic systems
+- **Filtering**: Uncertainty visualization, multi-target tracking
+- **Astronomy**: Ephemerides, orbital mechanics, relativistic corrections
+- **Navigation**: INS/GNSS integration, geophysical modeling
+- **Signal Processing**: Detection, filtering, transforms
+- **Terrain & Atmosphere**: Elevation models, atmospheric properties
+
+**Status**: ✅ All 29 examples validated and passing (100% execution success)
+
+### Tutorials (10 modules in `/docs/tutorials/`)
+
+Interactive learning modules with visualizations:
+- Assignment algorithms and 3D assignment problems
+- Atmospheric and geophysical models
+- Dynamical systems and reference frames
+- Filtering and smoothing techniques
+- Sensor fusion and advanced filtering
+- Special functions and mathematical tools
+
+**Status**: ✅ All 10 tutorials validated and passing (100% execution success)
+
+## Documentation
+
+- [API Reference](https://pytcl.readthedocs.io/en/latest/api/)
+- [User Guides](https://pytcl.readthedocs.io/en/latest/user_guide/)
+- [Examples](examples/) - 29 validated example scripts
+- [Tutorials](docs/tutorials/) - 10 interactive tutorial modules
 
 ## Comparison with Original MATLAB Library
 

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nrl-tracker"
-version = "1.7.0"
+version = "1.7.3"
 description = "Python port of the U.S. Naval Research Laboratory's Tracker Component Library for target tracking algorithms"
 readme = "README.md"
 authors = [

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pytcl/__init__.py

@@ -20,7 +20,7 @@ References
        no. 5, pp. 18-27, May 2017.
 """
 
-__version__ = "1.6.0"
+__version__ = "1.7.3"
 __author__ = "Python Port Contributors"
 __original_author__ = "David F. Crouse, Naval Research Laboratory"
 
@@ -44,7 +44,7 @@ from pytcl import (
 
 # Version tuple for programmatic access
 # Handle dev/alpha/beta/rc suffixes by extracting only numeric parts
-def _parse_version(version_str):
+def _parse_version(version_str: str) -> tuple[int, ...]:
     """Parse version string into tuple of integers."""
     import re
 

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pytcl/assignment_algorithms/gating.py

@@ -5,7 +5,7 @@ This module provides gating methods to determine which measurements
 fall within a validation region around predicted track states.
 """
 
-from typing import List, Tuple
+from typing import Any, List, Tuple
 
 import numpy as np
 from numba import njit
@@ -15,8 +15,8 @@ from scipy.stats import chi2
 
 @njit(cache=True, fastmath=True)
 def _mahalanobis_distance_2d(
-    innovation: np.ndarray,
-    S_inv: np.ndarray,
+    innovation: np.ndarray[Any, Any],
+    S_inv: np.ndarray[Any, Any],
 ) -> float:
     """JIT-compiled Mahalanobis distance for 2D innovations."""
     return innovation[0] * (
@@ -26,8 +26,8 @@ def _mahalanobis_distance_2d(
 
 @njit(cache=True, fastmath=True)
 def _mahalanobis_distance_3d(
-    innovation: np.ndarray,
-    S_inv: np.ndarray,
+    innovation: np.ndarray[Any, Any],
+    S_inv: np.ndarray[Any, Any],
 ) -> float:
     """JIT-compiled Mahalanobis distance for 3D innovations."""
     result = 0.0
@@ -39,8 +39,8 @@ def _mahalanobis_distance_3d(
 
 @njit(cache=True, fastmath=True)
 def _mahalanobis_distance_general(
-    innovation: np.ndarray,
-    S_inv: np.ndarray,
+    innovation: np.ndarray[Any, Any],
+    S_inv: np.ndarray[Any, Any],
 ) -> float:
     """JIT-compiled Mahalanobis distance for general dimension."""
     n = len(innovation)
@@ -341,9 +341,9 @@ def compute_gate_volume(
 
 @njit(cache=True, fastmath=True, parallel=False)
 def mahalanobis_batch(
-    innovations: np.ndarray,
-    S_inv: np.ndarray,
-    output: np.ndarray,
+    innovations: np.ndarray[Any, Any],
+    S_inv: np.ndarray[Any, Any],
+    output: np.ndarray[Any, Any],
 ) -> None:
     """
     Compute Mahalanobis distances for a batch of innovations.

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pytcl/assignment_algorithms/jpda.py

@@ -9,7 +9,7 @@ This is more sophisticated than GNN which makes hard assignment decisions,
 as JPDA can handle measurement origin uncertainty in cluttered environments.
 """
 
-from typing import List, NamedTuple, Optional, Tuple
+from typing import Any, List, NamedTuple, Optional
 
 import numpy as np
 from numba import njit
@@ -24,7 +24,7 @@ class JPDAResult(NamedTuple):
 
     Attributes
     ----------
-    association_probs : ndarray
+    association_probs : ndarray[Any]
         Association probability matrix of shape (n_tracks, n_measurements + 1).
         association_probs[i, j] is the probability that track i is associated
         with measurement j. The last column (j = n_measurements) represents
@@ -32,9 +32,9 @@ class JPDAResult(NamedTuple):
     marginal_probs : list of ndarray
         List of marginal association probabilities for each track.
         marginal_probs[i][j] = P(measurement j originated from track i).
-    likelihood_matrix : ndarray
+    likelihood_matrix : ndarray[Any]
         Measurement likelihood matrix of shape (n_tracks, n_measurements).
-    gated : ndarray
+    gated : ndarray[Any]
         Boolean matrix indicating which track-measurement pairs passed gating.
     """
 
@@ -53,7 +53,7 @@ class JPDAUpdate(NamedTuple):
         Updated state estimates for each track.
     covariances : list of ndarray
         Updated covariances for each track (includes spread of means).
-    association_probs : ndarray
+    association_probs : ndarray[Any]
         Association probability matrix.
     innovations : list of ndarray
         Combined weighted innovations for each track.
@@ -66,8 +66,8 @@ class JPDAUpdate(NamedTuple):
 
 
 def compute_measurement_likelihood(
-    innovation: NDArray,
-    innovation_cov: NDArray,
+    innovation: NDArray[Any],
+    innovation_cov: NDArray[Any],
     detection_prob: float = 1.0,
 ) -> float:
     """
@@ -75,9 +75,9 @@ def compute_measurement_likelihood(
 
     Parameters
     ----------
-    innovation : ndarray
+    innovation : ndarray[Any]
         Measurement innovation (residual), shape (m,).
-    innovation_cov : ndarray
+    innovation_cov : ndarray[Any]
         Innovation covariance, shape (m, m).
     detection_prob : float
         Probability of detection (Pd).
@@ -102,14 +102,14 @@ def compute_measurement_likelihood(
 
 
 def compute_likelihood_matrix(
-    track_states: List[NDArray],
-    track_covariances: List[NDArray],
-    measurements: NDArray,
-    H: NDArray,
-    R: NDArray,
+    track_states: list[NDArray[Any]],
+    track_covariances: list[NDArray[Any]],
+    measurements: NDArray[Any],
+    H: NDArray[Any],
+    R: NDArray[Any],
     detection_prob: float = 1.0,
     gate_threshold: Optional[float] = None,
-) -> Tuple[NDArray, NDArray]:
+) -> tuple[NDArray[Any], NDArray[Any]]:
     """
     Compute likelihood matrix for all track-measurement pairs.
 
@@ -119,11 +119,11 @@ def compute_likelihood_matrix(
         State estimates for each track.
     track_covariances : list of ndarray
         Covariances for each track.
-    measurements : ndarray
+    measurements : ndarray[Any]
         Measurements, shape (n_meas, m).
-    H : ndarray
+    H : ndarray[Any]
         Measurement matrix, shape (m, n).
-    R : ndarray
+    R : ndarray[Any]
         Measurement noise covariance, shape (m, m).
     detection_prob : float
         Probability of detection.
@@ -132,9 +132,9 @@ def compute_likelihood_matrix(
 
     Returns
     -------
-    likelihood_matrix : ndarray
+    likelihood_matrix : ndarray[Any]
         Likelihood values, shape (n_tracks, n_meas).
-    gated : ndarray
+    gated : ndarray[Any]
         Boolean gating matrix, shape (n_tracks, n_meas).
     """
     n_tracks = len(track_states)
@@ -163,11 +163,11 @@ def compute_likelihood_matrix(
 
 
 def jpda_probabilities(
-    likelihood_matrix: NDArray,
-    gated: NDArray,
+    likelihood_matrix: NDArray[Any],
+    gated: NDArray[Any],
     detection_prob: float = 1.0,
     clutter_density: float = 1e-6,
-) -> NDArray:
+) -> NDArray[Any]:
     """
     Compute JPDA association probabilities.
 
@@ -176,9 +176,9 @@ def jpda_probabilities(
 
     Parameters
     ----------
-    likelihood_matrix : ndarray
+    likelihood_matrix : ndarray[Any]
         Likelihood values, shape (n_tracks, n_meas).
-    gated : ndarray
+    gated : ndarray[Any]
         Boolean gating matrix, shape (n_tracks, n_meas).
     detection_prob : float
         Probability of detection (Pd).
@@ -187,7 +187,7 @@ def jpda_probabilities(
 
     Returns
     -------
-    beta : ndarray
+    beta : ndarray[Any]
         Association probability matrix, shape (n_tracks, n_meas + 1).
         beta[i, j] = P(measurement j is from track i) for j < n_meas.
         beta[i, n_meas] = P(track i has no measurement).
@@ -218,11 +218,11 @@ def jpda_probabilities(
 
 
 def _jpda_exact(
-    likelihood_matrix: NDArray,
-    gated: NDArray,
+    likelihood_matrix: NDArray[Any],
+    gated: NDArray[Any],
     detection_prob: float,
     clutter_density: float,
-) -> NDArray:
+) -> NDArray[Any]:
     """
     Exact JPDA computation via hypothesis enumeration.
 
@@ -241,8 +241,8 @@ def _jpda_exact(
     def generate_hypotheses(
         meas_idx: int,
         current_assignment: List[int],
-        used_tracks: set,
-    ):
+        used_tracks: set[Any],
+    ) -> Any:
         """Recursively generate valid hypotheses."""
         if meas_idx == n_meas:
             yield current_assignment.copy()
@@ -268,11 +268,11 @@ def _jpda_exact(
     hypothesis_probs = []
     hypothesis_assignments = []
 
-    for assignment in generate_hypotheses(0, [], set()):
+    for assignment in generate_hypotheses(0, [], set[Any]()):
         # Compute probability of this hypothesis
         prob = 1.0
 
-        detected_tracks = set()
+        detected_tracks = set[Any]()
         for j, track_idx in enumerate(assignment):
             if track_idx == -1:
                 # Measurement j is clutter
@@ -301,7 +301,7 @@ def _jpda_exact(
     for h_idx, (assignment, prob) in enumerate(
         zip(hypothesis_assignments, hypothesis_probs)
     ):
-        detected_tracks = set()
+        detected_tracks = set[Any]()
         for j, track_idx in enumerate(assignment):
             if track_idx >= 0:
                 beta[track_idx, j] += prob
@@ -317,11 +317,11 @@ def _jpda_exact(
 
 @njit(cache=True)
 def _jpda_approximate_core(
-    likelihood_matrix: np.ndarray,
-    gated: np.ndarray,
+    likelihood_matrix: np.ndarray[Any, Any],
+    gated: np.ndarray[Any, Any],
     detection_prob: float,
     clutter_density: float,
-) -> np.ndarray:
+) -> np.ndarray[Any, Any]:
     """JIT-compiled core of approximate JPDA computation."""
     n_tracks = likelihood_matrix.shape[0]
     n_meas = likelihood_matrix.shape[1]
@@ -369,11 +369,11 @@ def _jpda_approximate_core(
 
 
 def _jpda_approximate(
-    likelihood_matrix: NDArray,
-    gated: NDArray,
+    likelihood_matrix: NDArray[Any],
+    gated: NDArray[Any],
     detection_prob: float,
     clutter_density: float,
-) -> NDArray:
+) -> NDArray[Any]:
     """
     Approximate JPDA using parametric approach.
 

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pytcl/assignment_algorithms/nd_assignment.py

@@ -212,14 +212,15 @@ def relaxation_assignment_nd(
         result_relaxed = greedy_assignment_nd(relaxed_cost)
 
         # Compute lower bound from relaxed solution
-        lower_bound = (
-            result_relaxed.cost
-            + sum(np.sum(lambdas[d]) for d in range(n_dims))
+        lower_bound = result_relaxed.cost + sum(
+            np.sum(lambdas[d]) for d in range(n_dims)
         )
 
         # Extract solution from relaxed problem
         if len(result_relaxed.assignments) > 0:
-            actual_cost = float(np.sum(cost_tensor[tuple(result_relaxed.assignments.T)]))
+            actual_cost = float(
+                np.sum(cost_tensor[tuple(result_relaxed.assignments.T)])
+            )
 
             if actual_cost < best_cost:
                 best_cost = actual_cost

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pytcl/assignment_algorithms/network_flow.py

@@ -20,7 +20,7 @@ References
 """
 
 from enum import Enum
-from typing import NamedTuple, Tuple
+from typing import Any, NamedTuple, Tuple
 
 import numpy as np
 from numpy.typing import NDArray
@@ -79,7 +79,7 @@ class FlowEdge(NamedTuple):
 
 def assignment_to_flow_network(
     cost_matrix: NDArray[np.float64],
-) -> Tuple[list, NDArray, NDArray]:
+) -> Tuple[list[FlowEdge], NDArray[np.floating], NDArray[Any]]:
     """
     Convert 2D assignment problem to min-cost flow network.
 
@@ -140,7 +140,9 @@ def assignment_to_flow_network(
     # Tasks to sink: capacity 1, cost 0
     for j in range(1, n + 1):
         task_node = m + j
-        edges.append(FlowEdge(from_node=task_node, to_node=sink, capacity=1.0, cost=0.0))
+        edges.append(
+            FlowEdge(from_node=task_node, to_node=sink, capacity=1.0, cost=0.0)
+        )
 
     # Supply/demand: source supplies m units, sink demands m units
     supplies = np.zeros(n_nodes)
@@ -158,7 +160,7 @@ def assignment_to_flow_network(
 
 
 def min_cost_flow_successive_shortest_paths(
-    edges: list,
+    edges: list[FlowEdge],
     supplies: NDArray[np.float64],
     max_iterations: int = 1000,
 ) -> MinCostFlowResult:
@@ -260,7 +262,9 @@ def min_cost_flow_successive_shortest_paths(
 
         # Find minimum capacity along path
         min_flow = min(residual_capacity[e] for e in path_edges)
-        min_flow = min(min_flow, current_supplies[excess_node], -current_supplies[deficit_node])
+        min_flow = min(
+            min_flow, current_supplies[excess_node], -current_supplies[deficit_node]
+        )
 
         # Push flow along path
         total_cost = 0.0
@@ -295,7 +299,7 @@ def min_cost_flow_successive_shortest_paths(
 
 def assignment_from_flow_solution(
     flow: NDArray[np.float64],
-    edges: list,
+    edges: list[FlowEdge],
     cost_matrix_shape: Tuple[int, int],
 ) -> Tuple[NDArray[np.intp], float]:
     """
@@ -331,7 +335,11 @@ def assignment_from_flow_solution(
     assignment = np.array(assignment, dtype=np.intp)
     cost = 0.0
     if len(assignment) > 0:
-        cost = float(np.sum(flow[edge_idx] * edges[edge_idx].cost for edge_idx in range(len(edges))))
+        cost = float(
+            np.sum(
+                flow[edge_idx] * edges[edge_idx].cost for edge_idx in range(len(edges))
+            )
+        )
 
     return assignment, cost
 
@@ -356,6 +364,8 @@ def min_cost_assignment_via_flow(
     """
     edges, supplies, _ = assignment_to_flow_network(cost_matrix)
     result = min_cost_flow_successive_shortest_paths(edges, supplies)
-    assignment, cost = assignment_from_flow_solution(result.flow, edges, cost_matrix.shape)
+    assignment, cost = assignment_from_flow_solution(
+        result.flow, edges, cost_matrix.shape
+    )
 
     return assignment, cost

{nrl_tracker-1.7.0 → nrl_tracker-1.7.3}/pytcl/assignment_algorithms/three_dimensional/assignment.py

@@ -11,7 +11,7 @@ cost subject to the constraint that each index appears in at most one
 selected tuple.
 """
 
-from typing import List, NamedTuple, Optional, Tuple
+from typing import Any, List, NamedTuple, Optional, Tuple
 
 import numpy as np
 from numpy.typing import ArrayLike, NDArray
@@ -511,7 +511,7 @@ def assign3d_auction(
     assign_i: List[Optional[Tuple[int, int]]] = [None] * n1
 
     # Reverse: which i is assigned to (j, k)
-    reverse: dict = {}
+    reverse: dict[tuple[int, int], int] = {}
 
     converged = False
 
@@ -585,7 +585,7 @@ def assign3d(
     cost_tensor: ArrayLike,
     method: str = "lagrangian",
     maximize: bool = False,
-    **kwargs,
+    **kwargs: Any,
 ) -> Assignment3DResult:
     """
     Solve 3D assignment problem.