fix black

This commit is contained in:
yfyeung 2024-07-04 14:55:00 +08:00
parent 53da22ecc7
commit 7235b8561b

View File

@ -416,9 +416,7 @@ def decode_one_batch(
beam=params.beam_size,
)
else:
raise ValueError(
f"Unsupported decoding method: {params.decoding_method}"
)
raise ValueError(f"Unsupported decoding method: {params.decoding_method}")
hyps.append(sp.decode(hyp).split())
if params.decoding_method == "greedy_search":