diff --git a/airflow/models/taskinstance.py b/airflow/models/taskinstance.py index 79912a5f71a0e..6ff78617c5f1e 100644 --- a/airflow/models/taskinstance.py +++ b/airflow/models/taskinstance.py @@ -2765,7 +2765,9 @@ def _register_asset_changes_int( ) } if missing_assets := [ - unique_key.to_obj() for unique_key, _ in asset_alias_names if unique_key not in asset_models + asset_unique_key.to_asset() + for asset_unique_key, _ in asset_alias_names + if asset_unique_key not in asset_models ]: asset_models.update( (AssetUniqueKey.from_asset(asset_obj), asset_obj) diff --git a/airflow/utils/context.py b/airflow/utils/context.py index df4ecce89cb69..7681c9d6724df 100644 --- a/airflow/utils/context.py +++ b/airflow/utils/context.py @@ -208,7 +208,9 @@ def __str__(self) -> str: return f"OutletEventAccessors(_dict={self._dict})" def __iter__(self) -> Iterator[Asset | AssetAlias]: - return iter(key.to_obj() if isinstance(key, BaseAssetUniqueKey) else key for key in self._dict) + return iter( + key.to_asset() if isinstance(key, AssetUniqueKey) else key.to_asset_alias() for key in self._dict + ) def __len__(self) -> int: return len(self._dict) diff --git a/task_sdk/src/airflow/sdk/definitions/asset/__init__.py b/task_sdk/src/airflow/sdk/definitions/asset/__init__.py index 7abc4e53d7472..7f2adfdc9a444 100644 --- a/task_sdk/src/airflow/sdk/definitions/asset/__init__.py +++ b/task_sdk/src/airflow/sdk/definitions/asset/__init__.py @@ -58,7 +58,7 @@ class AssetUniqueKey(NamedTuple): def from_asset(asset: Asset | AssetModel) -> AssetUniqueKey: return AssetUniqueKey(name=asset.name, uri=asset.uri) - def to_obj(self) -> Asset: + def to_asset(self) -> Asset: return Asset(name=self.name, uri=self.uri) @@ -69,7 +69,7 @@ class AssetAliasUniqueKey(NamedTuple): def from_asset_alias(asset_alias: AssetAlias) -> AssetAliasUniqueKey: return AssetAliasUniqueKey(name=asset_alias.name) - def to_obj(self) -> AssetAlias: + def to_asset_alias(self) -> AssetAlias: return AssetAlias(name=self.name)