allow Chip or XY as key

SpiNNakerManchester · Mar 5, 2024 · a5dd539 · a5dd539
1 parent d937fd9
commit a5dd539
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/spinn_machine/machine.py b/spinn_machine/machine.py
@@ -30,8 +30,8 @@
     from .chip import Chip
 
 logger = logging.getLogger(__name__)
-_SpinLinkKey: TypeAlias = Tuple[Union[str, XY], int]
-_FpgaLinkKey: TypeAlias = Tuple[Union[str, XY], int, int]
+_SpinLinkKey: TypeAlias = Tuple[Union[str, XY, "Chip"], int]
+_FpgaLinkKey: TypeAlias = Tuple[Union[str, XY, "Chip"], int, int]
 
 
 class Machine(object, metaclass=AbstractBase):