add type annotation

pixee · Nov 22, 2023 · a54f36b · a54f36b
1 parent 5e173f2
commit a54f36b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/codemodder/dependency_management/setup_py_codemod.py b/src/codemodder/dependency_management/setup_py_codemod.py
@@ -6,6 +6,7 @@
 from codemodder.codemods.utils import is_setup_py_file
 from codemodder.codemods.utils_mixin import NameResolutionMixin
 from codemodder.file_context import FileContext
+from packaging.requirements import Requirement
 
 
 class SetupPyAddDependencies(BaseCodemod, NameResolutionMixin):
@@ -16,7 +17,10 @@ class SetupPyAddDependencies(BaseCodemod, NameResolutionMixin):
     REFERENCES: list = []
 
     def __init__(
-        self, codemod_context: CodemodContext, file_context: FileContext, dependencies
+        self,
+        codemod_context: CodemodContext,
+        file_context: FileContext,
+        dependencies: list[Requirement],
     ):
         BaseCodemod.__init__(self, codemod_context, file_context)
         NameResolutionMixin.__init__(self)