flake8 cleanup

pull/380/head
evelynmitchell 1 year ago
parent fb1d72a6d1
commit d95ee67fbf

@ -245,10 +245,7 @@ class SamplingParams:
"early_stopping is not effective and must be "
"False when not using beam search."
)
if (
self.length_penalty < 1.0 - _SAMPLING_EPS
or self.length_penalty > 1.0 + _SAMPLING_EPS
):
if (self.length_penalty < 1.0 - _SAMPLING_EPS or self.length_penalty > 1.0 + _SAMPLING_EPS):
raise ValueError(
"length_penalty is not effective and must be the "
"default value of 1.0 when not using beam search."

Loading…
Cancel
Save