Enhance displaying and debuggability in dump comparator (#19466)
This commit is contained in:
@@ -124,7 +124,6 @@ class TestExecuteAlignerPlan:
|
||||
y=[self._make_step_plan(step=0, indices=[0])],
|
||||
),
|
||||
token_aligner_plan=None,
|
||||
token_dims=Pair(x=0, y=0),
|
||||
)
|
||||
|
||||
tensors_pair: Pair[list[torch.Tensor]] = Pair(
|
||||
@@ -146,7 +145,6 @@ class TestExecuteAlignerPlan:
|
||||
y=[self._make_step_plan(step=0, indices=[0, 1])],
|
||||
),
|
||||
token_aligner_plan=None,
|
||||
token_dims=Pair(x=0, y=0),
|
||||
)
|
||||
|
||||
tensors_pair: Pair[list[torch.Tensor]] = Pair(
|
||||
@@ -168,7 +166,6 @@ class TestExecuteAlignerPlan:
|
||||
y=[self._make_step_plan(step=0, indices=[0])],
|
||||
),
|
||||
token_aligner_plan=None,
|
||||
token_dims=Pair(x=0, y=0),
|
||||
)
|
||||
|
||||
t_x: torch.Tensor = torch.tensor([1.0, 2.0])
|
||||
@@ -191,7 +188,6 @@ class TestExecuteAlignerPlan:
|
||||
y=[self._make_step_plan(step=0, indices=[0])],
|
||||
),
|
||||
token_aligner_plan=None,
|
||||
token_dims=Pair(x=0, y=0),
|
||||
)
|
||||
|
||||
tensors_pair: Pair[list[torch.Tensor]] = Pair(
|
||||
|
||||
Reference in New Issue
Block a user