From dcb616f54592f688fbd38dc84fcf3f63281f0dae Mon Sep 17 00:00:00 2001 From: Shahar Epstein <60007259+shahar1@users.noreply.github.com> Date: Fri, 29 Nov 2024 23:25:16 +0200 Subject: [PATCH] Remove internal api call decorator from manager.py --- airflow/assets/manager.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/airflow/assets/manager.py b/airflow/assets/manager.py index 0616d60151130..40bc97b8134c7 100644 --- a/airflow/assets/manager.py +++ b/airflow/assets/manager.py @@ -23,7 +23,6 @@ from sqlalchemy import exc, select from sqlalchemy.orm import joinedload -from airflow.api_internal.internal_api_call import internal_api_call from airflow.configuration import conf from airflow.listeners.listener import get_listener_manager from airflow.models.asset import ( @@ -35,7 +34,6 @@ DagScheduleAssetReference, ) from airflow.models.dagbag import DagPriorityParsingRequest -from airflow.sdk.definitions.asset import Asset from airflow.stats import Stats from airflow.utils.log.logging_mixin import LoggingMixin @@ -104,7 +102,6 @@ def _add_asset_alias_association( ) @classmethod - @internal_api_call def register_asset_change( cls, *, @@ -122,7 +119,6 @@ def register_asset_change( For local assets, look them up, record the asset event, queue dagruns, and broadcast the asset event """ - # todo: add test so that all usages of internal_api_call are added to rpc endpoint asset_model = session.scalar( select(AssetModel) .where(AssetModel.uri == asset.uri)