Use Optional

sebastinas · sebastinas · commit 7e64b0f41233 · 2022-09-25T20:34:12.000+02:00
diff --git a/bpython/importcompletion.py b/bpython/importcompletion.py
@@ -155,9 +155,7 @@ def complete(self, cursor_offset: int, line: str) -> Optional[Set[str]]:
         else:
             return None
 
-    def find_modules(
-        self, path: Path
-    ) -> Generator[Union[str, None], None, None]:
+    def find_modules(self, path: Path) -> Generator[Optional[str], None, None]:
         """Find all modules (and packages) for a given directory."""
         if not path.is_dir():
             # Perhaps a zip file
