Skip to content

Commit 1735495

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 467b7a1 commit 1735495

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

tests/_runtime/test_dataflow.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1445,7 +1445,9 @@ def test_prune_cells_for_overrides_single_cell() -> None:
14451445
execution_order = ["0", "1", "2"]
14461446

14471447
# Override x - should prune cell 0
1448-
result = dataflow.prune_cells_for_overrides(graph, execution_order, {"x": 100})
1448+
result = dataflow.prune_cells_for_overrides(
1449+
graph, execution_order, {"x": 100}
1450+
)
14491451
assert result == ["1", "2"]
14501452

14511453

@@ -1537,7 +1539,9 @@ def test_prune_cells_for_overrides_partial_override() -> None:
15371539
execution_order = ["0", "1", "2"]
15381540

15391541
# Override only x - should prune only cell 0
1540-
result = dataflow.prune_cells_for_overrides(graph, execution_order, {"x": 100})
1542+
result = dataflow.prune_cells_for_overrides(
1543+
graph, execution_order, {"x": 100}
1544+
)
15411545
assert result == ["1", "2"]
15421546

15431547

@@ -1581,5 +1585,7 @@ def test_prune_cells_for_overrides_preserves_order() -> None:
15811585
execution_order = ["0", "1", "2", "3"]
15821586

15831587
# Override b - should prune only cell 1, preserving order
1584-
result = dataflow.prune_cells_for_overrides(graph, execution_order, {"b": 100})
1588+
result = dataflow.prune_cells_for_overrides(
1589+
graph, execution_order, {"b": 100}
1590+
)
15851591
assert result == ["0", "2", "3"]

0 commit comments

Comments
 (0)