@@ -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