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)