From a54f36b9ceb91329603f904ba045df7a1b861d4d Mon Sep 17 00:00:00 2001 From: clavedeluna Date: Wed, 22 Nov 2023 07:38:17 -0300 Subject: [PATCH] add type annotation --- src/codemodder/dependency_management/setup_py_codemod.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/codemodder/dependency_management/setup_py_codemod.py b/src/codemodder/dependency_management/setup_py_codemod.py index 5c51fcd2..0f3a239a 100644 --- 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)