jan/masque
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix type annotations for Device.__getitem__

Browse Source
This commit is contained in:
Jan Petykiewicz 2021-06-22 18:25:31 -07:00
parent 65466a5986
commit 3aefa644e5
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
masque/builder/devices.py
View File

@ -1,4 +1,5 @@
from typing import Dict, Iterable, List, Tuple, Union, TypeVar, Any, Iterator, Optional, Sequence from typing import Dict, Iterable, List, Tuple, Union, TypeVar, Any, Iterator, Optional, Sequence
from typing import overload, KeysView, ValuesView
import copy import copy
import warnings import warnings
import traceback import traceback
@ -196,7 +197,15 @@ class Device(Copyable, Mirrorable):
self._dead = False self._dead = False
def __getitem__(self, key: Union[str, Iterable[str]]) -> numpy.ndarray: @overload
def __getitem__(self, key: str) -> Port:
pass
@overload
def __getitem__(self, key: Union[List[str], Tuple[str], KeysView[str], ValuesView[str]]) -> Dict[str, Port]:
pass
def __getitem__(self, key: Union[str, Iterable[str]]) -> Union[Port, Dict[str, Port]]:
""" """
For convenience, ports can be read out using square brackets: For convenience, ports can be read out using square brackets:
- `device['A'] == Port((0, 0), 0)` - `device['A'] == Port((0, 0), 0)`