diff --git a/providers/papermill/src/airflow/providers/papermill/operators/papermill.py b/providers/papermill/src/airflow/providers/papermill/operators/papermill.py index 5229f743a58d8..e02f259b60475 100644 --- a/providers/papermill/src/airflow/providers/papermill/operators/papermill.py +++ b/providers/papermill/src/airflow/providers/papermill/operators/papermill.py @@ -34,7 +34,7 @@ from airflow.sdk.definitions.context import Context except ImportError: # TODO: Remove once provider drops support for Airflow 2 - from airflow.utils.context import Context # type: ignore[no-redef] + from airflow.utils.context import Context @attr.s(auto_attribs=True) @@ -106,9 +106,9 @@ def __init__( def execute(self, context: Context): if not isinstance(self.input_nb, NoteBook): - self.input_nb = NoteBook(url=self.input_nb, parameters=self.parameters) # type: ignore[call-arg] + self.input_nb = NoteBook(url=self.input_nb, parameters=self.parameters) if not isinstance(self.output_nb, NoteBook): - self.output_nb = NoteBook(url=self.output_nb) # type: ignore[call-arg] + self.output_nb = NoteBook(url=self.output_nb) if not AIRFLOW_V_3_0_PLUS: self.inlets.append(self.input_nb) self.outlets.append(self.output_nb) diff --git a/providers/papermill/src/airflow/providers/papermill/version_compat.py b/providers/papermill/src/airflow/providers/papermill/version_compat.py index 7c415980c9307..2b7353188ec58 100644 --- a/providers/papermill/src/airflow/providers/papermill/version_compat.py +++ b/providers/papermill/src/airflow/providers/papermill/version_compat.py @@ -33,7 +33,7 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]: if AIRFLOW_V_3_0_PLUS: from airflow.sdk import BaseOperator else: - from airflow.models import BaseOperator # type: ignore[no-redef] + from airflow.models import BaseOperator if AIRFLOW_V_3_1_PLUS: from airflow.sdk import BaseHook