From 8686c71ccf982b9f552a258148c1e08b3da01c4b Mon Sep 17 00:00:00 2001 From: gmega Date: Thu, 31 Oct 2024 15:26:00 -0300 Subject: [PATCH] fix typing --- benchmarks/experiments/tests/test_static_experiment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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