Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] authored and GNroy committed Oct 5, 2023
1 parent 8e6244b commit c9f8a84
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/collections/asr/test_asr_rnnt_encdec_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,9 @@ def joint(self, f: torch.Tensor, g: torch.Tensor) -> torch.Tensor:
setup["joint"] = DummyRNNTJoint()
# expected timesteps for max_symbols_per_step=5 are [[0, 0, 0, 0, 0, 1, 1], [1, 1, 1, 1, 1]],
# so we have both looped and regular iteration on the second frame
setup["encoder_output"] = torch.tensor([[[1, 0, 0], [0, 1, 0], [0, 0, 1]], [[0, 0, 1], [2, 0, 0], [0, 0, 0]]], dtype=torch.float32).transpose(1, 2)
setup["encoder_output"] = torch.tensor(
[[[1, 0, 0], [0, 1, 0], [0, 0, 1]], [[0, 0, 1], [2, 0, 0], [0, 0, 0]]], dtype=torch.float32
).transpose(1, 2)
setup["encoded_lengths"] = torch.tensor([3, 2])
return setup

Expand Down

0 comments on commit c9f8a84

Please sign in to comment.