Skip to content

Commit 39c80c7

Browse files
committed
Add Missing Type Annotations
1 parent 5dfabcb commit 39c80c7

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

adafruit_trellis.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ class TrellisLEDs:
107107
def __init__(self, trellis_obj: "Trellis") -> None:
108108
self._parent = trellis_obj
109109

110-
def __getitem__(self, x: int) -> None:
110+
def __getitem__(self, x: int) -> bool:
111111
if 0 < x >= self._parent._num_leds:
112112
raise ValueError(
113113
("LED number must be between 0 -", self._parent._num_leds - 1)
@@ -175,7 +175,7 @@ class Trellis:
175175
176176
"""
177177

178-
def __init__(self, i2c: I2C, addresses: Optional[int] = None) -> None:
178+
def __init__(self, i2c: I2C, addresses: Optional[List[int]] = None) -> None:
179179
if addresses is None:
180180
addresses = [0x70]
181181
self._i2c_devices = []
@@ -211,7 +211,7 @@ def _write_cmd(self, byte: int) -> None:
211211
device.write(self._temp)
212212

213213
@property
214-
def blink_rate(self) -> int:
214+
def blink_rate(self) -> Literal[0, 1, 2, 3]:
215215
"""
216216
The current blink rate as an integer range 0-3.
217217
"""
@@ -233,9 +233,7 @@ def brightness(self) -> int:
233233
return self._brightness
234234

235235
@brightness.setter
236-
def brightness(
237-
self, brightness: Literal[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
238-
) -> None:
236+
def brightness(self, brightness: int) -> None:
239237
if not 0 <= brightness <= 15:
240238
raise ValueError("Brightness must be an integer in the range: 0-15")
241239
brightness = brightness & 0x0F

0 commit comments

Comments
 (0)