Skip to content

Commit

Permalink
apply black
Browse files Browse the repository at this point in the history
  • Loading branch information
ftshijt committed Apr 27, 2022
1 parent 04d0cd8 commit c4b93e8
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions test/espnet2/bin/test_st_inference.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,15 +65,11 @@ def st_config_file(tmp_path: Path, token_list, src_token_list):

@pytest.mark.execution_timeout(5)
def test_Speech2Text(st_config_file):
speech2text = Speech2Text(
st_train_config=st_config_file, beam_size=1
)
speech2text = Speech2Text(st_train_config=st_config_file, beam_size=1)
speech = np.random.randn(1000)
results = speech2text(speech)
for text, token, token_int, hyp in results:
assert isinstance(text, str)
assert isinstance(token[0], str)
assert isinstance(token_int[0], int)
assert isinstance(hyp, Hypothesis)


0 comments on commit c4b93e8

Please sign in to comment.