Skip to content

Commit dd90fc6

Browse files
Fix HyperCommGrid test formatting
1 parent 644aab2 commit dd90fc6

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

‎tests/unit_tests/test_hyper_comm_grid.py‎

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -430,9 +430,7 @@ def test_register_view_shared_dim_membership_match(self):
430430
# base: tp=2, dp=2, pp=2 -> reversed [dp, tp, ...]; pp trailing -> contiguous blocks.
431431
grid = HyperCommGrid([2, 2, 2], ["tp", "dp", "pp"])
432432
# expert: expt_tp=2, ep=2, pp=2 -> pp trailing -> same membership as base.
433-
grid.register_view(
434-
"expert", [2, 2, 2], ["expt_tp", "ep", "pp"], shared_dims=["pp"]
435-
)
433+
grid.register_view("expert", [2, 2, 2], ["expt_tp", "ep", "pp"], shared_dims=["pp"])
436434

437435
assert grid._views["expert"].shared_dims == ["pp"]
438436
# Both layouts enumerate pp identically.

0 commit comments

Comments
 (0)