fix(azure): remove deprecated pipelineinfo

astronomer · Oct 18, 2023 · 0165bff · 0165bff
1 parent 9677f74
commit 0165bff
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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,