diff --git a/openpilot/starpilot/flm/tests/test_tracking_pose_interval.py b/openpilot/starpilot/flm/tests/test_tracking_pose_interval.py index 2eaa4b779a..fc9529b9b1 100644 --- a/openpilot/starpilot/flm/tests/test_tracking_pose_interval.py +++ b/openpilot/starpilot/flm/tests/test_tracking_pose_interval.py @@ -1,6 +1,11 @@ """Pair selection only; these operands grant no native or application authority.""" from types import SimpleNamespace import unittest +from typing import TYPE_CHECKING, cast + +if TYPE_CHECKING: + from openpilot.starpilot.flm.live_tracking import LiveTrackingOptimizer + from openpilot.starpilot.flm.live_tracking_episodes import TrackingEpisodeBuilder from openpilot.starpilot.flm.live_tracking_ingress import LiveTrackingIngress @@ -9,8 +14,8 @@ def subject(source_tracking=True): owner = object.__new__(LiveTrackingIngress) owner.binding = SimpleNamespace(identity='1'*64, source_epoch='2'*64, baseline_revision='3'*64, controller_revision='revision', native_identity='4'*64, native_required=True) - owner.episodes = SimpleNamespace(source_tracking=source_tracking) - owner.optimizer = SimpleNamespace(lag_ns=0) + owner.episodes = cast("TrackingEpisodeBuilder", SimpleNamespace(source_tracking=source_tracking)) + owner.optimizer = cast("LiveTrackingOptimizer", SimpleNamespace(lag_ns=0)) owner.last_coordinates = owner.last_profile = None return owner @@ -126,7 +131,6 @@ class TestSourceExperimentEntry(unittest.TestCase): def test_source_experiment_still_allocates_three_fresh_baseline_windows(self): from openpilot.starpilot.flm.live_tracking import LiveTrackingOptimizer - import inspect # Exercise the real allocation without manufacturing source/native authority. function = next(value for name, value in vars(LiveTrackingOptimizer).items() if name == '_comparison_allocation') scope = dict(function.__globals__) @@ -136,5 +140,6 @@ class TestSourceExperimentEntry(unittest.TestCase): owner = SimpleNamespace(rmse=None, binding='binding', accepted=dict.fromkeys(scope['NAMES'], 0.), lag_ns=0) owner.planning_history = [SimpleNamespace(binding='binding', profile_sha256='profile', end_ns=n*2_000_000_000) for n in (1,2,3)] episode = SimpleNamespace(kind='subtle', profile_sha256='profile', end_ns=6_000_000_000) - target = dict(owner.accepted); target[next(iter(target))] = .005 + target = dict(owner.accepted) + target[next(iter(target))] = .005 assert allocation(owner, episode, .001, target, minimum_episodes=1) == (3,3) diff --git a/prebuilt.json b/prebuilt.json index c41da68400..cc79b80f72 100644 --- a/prebuilt.json +++ b/prebuilt.json @@ -1 +1 @@ -{"protected_inventory_sha256": "a9dba7eeadb24e9130561710b2a2592a6363732f00dc487f61fc06d2243de667", "schema_version": 1} +{"protected_inventory_sha256": "d03437ef0111e5dbc4723a64087fa3c0c74139f52bf657faf1d159d524ba43f4", "schema_version": 1}