Skip to content

Commit

Permalink
ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
schrockn committed Sep 10, 2023
1 parent f99cf65 commit 863fa1b
Showing 1 changed file with 3 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -1078,14 +1078,14 @@ def asset_partitions_subset_for_input(
if upstream_asset_key is not None:
check.failed("The input has no asset partitions")

assert upstream_asset_key # appease pyright
assert upstream_asset_key # appease pyright

upstream_asset_partitions_def = asset_layer.partitions_def_for_asset(upstream_asset_key)

if upstream_asset_partitions_def is None:
check.failed("The input has no asset partitions")

assert upstream_asset_partitions_def # appease pyright
assert upstream_asset_partitions_def # appease pyright

partitions_def = assets_def.partitions_def if assets_def else None
partitions_subset = (
Expand All @@ -1096,9 +1096,7 @@ def asset_partitions_subset_for_input(
else None
)
partition_mapping = infer_partition_mapping(
asset_layer.partition_mapping_for_node_input(
self.node_handle, upstream_asset_key
),
asset_layer.partition_mapping_for_node_input(self.node_handle, upstream_asset_key),
partitions_def,
upstream_asset_partitions_def,
)
Expand All @@ -1123,7 +1121,6 @@ def asset_partitions_subset_for_input(

return mapped_partitions_result.partitions_subset


def asset_partition_key_for_input(self, input_name: str) -> str:
start, end = self.asset_partition_key_range_for_input(input_name)
if start == end:
Expand Down

0 comments on commit 863fa1b

Please sign in to comment.