From cb0cbbdeedcc0df856e93f8ae226f75ef910897c Mon Sep 17 00:00:00 2001 From: kalyanr Date: Sun, 1 Dec 2024 09:32:16 +0530 Subject: [PATCH] remove internal_api_call decorator --- providers/src/airflow/providers/edge/models/edge_job.py | 3 --- providers/src/airflow/providers/edge/models/edge_logs.py | 2 -- 2 files changed, 5 deletions(-) diff --git a/providers/src/airflow/providers/edge/models/edge_job.py b/providers/src/airflow/providers/edge/models/edge_job.py index c591b6d30529e..bd4b3b191c1b9 100644 --- a/providers/src/airflow/providers/edge/models/edge_job.py +++ b/providers/src/airflow/providers/edge/models/edge_job.py @@ -30,7 +30,6 @@ text, ) -from airflow.api_internal.internal_api_call import internal_api_call from airflow.models.base import Base, StringID from airflow.models.taskinstancekey import TaskInstanceKey from airflow.serialization.serialized_objects import add_pydantic_class_type_mapping @@ -127,7 +126,6 @@ def key(self) -> TaskInstanceKey: return TaskInstanceKey(self.dag_id, self.task_id, self.run_id, self.try_number, self.map_index) @staticmethod - @internal_api_call @provide_session def reserve_task( worker_name: str, @@ -170,7 +168,6 @@ def reserve_task( ) @staticmethod - @internal_api_call @provide_session def set_state(task: TaskInstanceKey | tuple, state: TaskInstanceState, session: Session = NEW_SESSION): if isinstance(task, tuple): diff --git a/providers/src/airflow/providers/edge/models/edge_logs.py b/providers/src/airflow/providers/edge/models/edge_logs.py index 65146cf7edc3b..d5dfac4d09c92 100644 --- a/providers/src/airflow/providers/edge/models/edge_logs.py +++ b/providers/src/airflow/providers/edge/models/edge_logs.py @@ -30,7 +30,6 @@ ) from sqlalchemy.dialects.mysql import MEDIUMTEXT -from airflow.api_internal.internal_api_call import internal_api_call from airflow.configuration import conf from airflow.models.base import Base, StringID from airflow.models.taskinstance import TaskInstance @@ -99,7 +98,6 @@ class EdgeLogs(BaseModel, LoggingMixin): model_config = ConfigDict(from_attributes=True, arbitrary_types_allowed=True) @staticmethod - @internal_api_call @provide_session def push_logs( task: TaskInstanceKey | tuple,