Fix type errors

facebook · Aug 21, 2023 · de998fd · de998fd
1 parent 12d9935
commit de998fd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/client/configuration/search_path.py b/client/configuration/search_path.py
@@ -28,9 +28,9 @@
 LOG: logging.Logger = logging.getLogger(__name__)
 
 dist_info_in_root: Dict[str, List[str]] = {}
-_site_filter = re.compile(r".*-([0-99]\.)*dist-info")
+_site_filter: re.Pattern[str] = re.compile(r".*-([0-99]\.)*dist-info")
 
-_PYCACHE = re.compile("__pycache__")
+_PYCACHE: re.Pattern[str] = re.compile("__pycache__")
 
 
 def _expand_relative_root(path: str, relative_root: str) -> str:
@@ -99,7 +99,10 @@ def package_path(self) -> Union[str, None]:
         else:
             return None
 
-        not_toplevel_patterns: Tuple[re.Pattern] = (this_pkg_filter, _PYCACHE)
+        not_toplevel_patterns: Tuple[re.Pattern[str], re.Pattern[str]] = (
+            this_pkg_filter,
+            _PYCACHE,
+        )
 
         with open(file=f"{dist_info_path}/RECORD", mode="r") as record:
             files = []

diff --git a/client/configuration/tests/search_path_test.py b/client/configuration/tests/search_path_test.py
@@ -250,7 +250,7 @@ def test_process_required_raw_elements_site_package_nonexistence(self) -> None:
                 required=True,
             )
 
-    def test_toplevel_module_not_pyfile(self):
+    def test_toplevel_module_not_pyfile(self) -> None:
         Path.mkdir(Path("foo"), exist_ok=True)
         Path.mkdir(Path("foo/bar-1.0.0.dist-info"), exist_ok=True)
         Path.touch(Path("foo/bar.so"), exist_ok=True)