diff --git a/benchmarks/experiments/tests/test_static_experiment.py b/benchmarks/experiments/tests/test_static_experiment.py index b609902..00262c4 100644 --- a/benchmarks/experiments/tests/test_static_experiment.py +++ b/benchmarks/experiments/tests/test_static_experiment.py @@ -19,7 +19,7 @@ def mock_sampler(elements: List[int]) -> Sampler: class MockNode(Node[MockHandle]): def __init__(self): - self.seeding: Optional[MockNode] = None + self.seeding: Optional[Path] = None self.leeching: Optional[MockHandle] = None def seed(self, path: Path, handle: Optional[MockHandle] = None) -> MockHandle: @@ -36,7 +36,7 @@ class MockFileSharingNetwork(FileSharingNetwork[MockNode]): self._nodes = [MockNode() for _ in range(n)] @property - def nodes(self) -> List[TNode]: + def nodes(self) -> List[MockNode]: return self._nodes