Skip to content

Commit 4f7d47c

Browse files
committed
fixup! fixup! fixup! fixup! fixup! tmp
1 parent b2bdd5d commit 4f7d47c

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

airflow-core/src/airflow/assets/manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ def _queue_partitioned_dags(
391391
if TYPE_CHECKING:
392392
assert isinstance(timetable, PartitionedAssetTimetable)
393393

394-
if asset_model := session.scalar(select(AssetModel).where(AssetModel.id == asset_id)) is None:
394+
if (asset_model := session.scalar(select(AssetModel).where(AssetModel.id == asset_id))) is None:
395395
raise ValueError()
396396

397397
target_key = timetable.get_partition_mapper(

airflow-core/src/airflow/timetables/simple.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -267,14 +267,14 @@ def serialize(self) -> dict[str, Any]:
267267

268268
return {
269269
"asset_condition": encode_asset_like(self.asset_condition),
270-
"partition_mapper": self.default_partition_mapper.serialize(),
270+
"default_partition_mapper": self.default_partition_mapper.serialize(),
271271
"_partition_mappers_by_name": [
272272
(name, encode_partition_mapper(partition_mapper))
273273
for name, partition_mapper in self._partition_mappers_by_name.items()
274274
],
275275
"_partition_mappers_by_uri": [
276276
(uri, encode_partition_mapper(partition_mapper))
277-
for uri, partition_mapper in self._partition_mappers_by_ui.items()
277+
for uri, partition_mapper in self._partition_mappers_by_uri.items()
278278
],
279279
}
280280

@@ -285,7 +285,7 @@ def deserialize(cls, data: dict[str, Any]) -> Timetable:
285285

286286
timetable = cls(
287287
assets=decode_asset_like(data["asset_condition"]),
288-
default_partition_mapper=decode_partition_mapper(data["partition_mapper"]),
288+
default_partition_mapper=decode_partition_mapper(data["default_partition_mapper"]),
289289
)
290290
timetable._partition_mappers_by_name = {
291291
name: decode_partition_mapper(ser_partition_mapper)

0 commit comments

Comments
 (0)