add type annotations

pdm-project · Mar 24, 2021 · b0f9c91 · b0f9c91
1 parent 4251e51
commit b0f9c91
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pdm/models/setup.py b/pdm/models/setup.py
@@ -216,7 +216,7 @@ def _find_sub_setup_call(
 
     @classmethod
     def _find_install_requires(cls, call: ast.Call, body: Iterable[Any]) -> List[str]:
-        install_requires = []
+        install_requires: List[str] = []
         value = cls._find_in_call(call, "install_requires")
         if value is None:
             # Trying to find in kwargs
@@ -258,8 +258,8 @@ def _find_install_requires(cls, call: ast.Call, body: Iterable[Any]) -> List[str
     @classmethod
     def _find_extras_require(
         cls, call: ast.Call, body: Iterable[Any]
-    ) -> Dict[str, List]:
-        extras_require = {}
+    ) -> Dict[str, List[str]]:
+        extras_require: Dict[str, List[str]] = {}
         value = cls._find_in_call(call, "extras_require")
         if value is None:
             # Trying to find in kwargs