14

tests/test_planner_revision.py

@@ -5,12 +5,13 @@ import math
 import pytest
 
 from mcm26b.config import ProblemConfig
+from mcm26b.harness import LocalSimTransport, LocalSimulator
 from mcm26b.planner import ClearCertificate, DirectionalExtension
 from mcm26b.planner.bayes_loop import BayesLoop, Task
 from mcm26b.planner.seven import validate_seven_coverage
 from mcm26b.protocol.client import SimulatorClient
 from mcm26b.protocol.rules import DIRECTION, NO_SIGNAL
-from mcm26b.sim import LocalSimulator, LocalSimTransport, Source
+from referee import Source
 
 
 def setup(sources=(), cls=BayesLoop, config=ProblemConfig()):