From c1e5e49e8fbd125a3bdd3b9ed260917b572daa9c Mon Sep 17 00:00:00 2001 From: Thomas Franks Date: Mon, 19 Sep 2022 14:51:28 -0400 Subject: [PATCH] Add Missing Type Annotations --- adafruit_bitbangio.py | 12 +++++++++--- requirements.txt | 2 +- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/adafruit_bitbangio.py b/adafruit_bitbangio.py index 551f245..a10d957 100644 --- a/adafruit_bitbangio.py +++ b/adafruit_bitbangio.py @@ -24,8 +24,9 @@ """ try: - from typing import Optional, List + from typing import List, Optional, Type from typing_extensions import Literal + from types import TracebackType from circuitpython_typing import WriteableBuffer, ReadableBuffer from microcontroller import Pin except ImportError: @@ -67,10 +68,15 @@ def _check_lock(self) -> Literal[True]: raise RuntimeError("First call try_lock()") return True - def __enter__(self): + def __enter__(self) -> "_BitBangIO": return self - def __exit__(self, exc_type, exc_value, traceback) -> None: + def __exit__( + self, + exc_type: Optional[Type[BaseException]], + exc_value: Optional[BaseException], + traceback: Optional[TracebackType], + ) -> None: self.deinit() # pylint: disable=no-self-use diff --git a/requirements.txt b/requirements.txt index 452ef26..b860c77 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,4 +3,4 @@ # SPDX-License-Identifier: Unlicense Adafruit-Blinka -typing-extensions +typing-extensions~=4.0