diff --git a/astronomer/providers/microsoft/azure/operators/data_factory.py b/astronomer/providers/microsoft/azure/operators/data_factory.py index b9a2b6eb8..85bcdf2e1 100644 --- a/astronomer/providers/microsoft/azure/operators/data_factory.py +++ b/astronomer/providers/microsoft/azure/operators/data_factory.py @@ -6,7 +6,6 @@ AzureDataFactoryHook, AzureDataFactoryPipelineRunException, AzureDataFactoryPipelineRunStatus, - PipelineRunInfo, ) from airflow.providers.microsoft.azure.operators.data_factory import ( AzureDataFactoryRunPipelineOperator, @@ -67,12 +66,11 @@ def execute(self, context: Context) -> None: context["ti"].xcom_push(key="run_id", value=run_id) end_time = time.time() + self.timeout - pipeline_run_info = PipelineRunInfo( + pipeline_run_status = hook.get_pipeline_run_status( run_id=run_id, - factory_name=self.factory_name, resource_group_name=self.resource_group_name, + factory_name=self.factory_name, ) - pipeline_run_status = hook.get_pipeline_run_status(**pipeline_run_info) if pipeline_run_status not in AzureDataFactoryPipelineRunStatus.TERMINAL_STATUSES: self.defer( timeout=self.execution_timeout,