Skip to content

GitLab

Commit 91a06081 authored by baberabb's avatar baberabb
Browse files

fixup! added tolerance for likelihood, rolling

parent dbbc613f
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/test_huggingface.py
View file @ 91a06081
...@@ -90,7 +90,7 @@ class Test_HFLM: ...@@ -90,7 +90,7 @@ class Test_HFLM:
def test_logliklihood(self) -> None: def test_logliklihood(self) -> None:
res = self.LM.loglikelihood(self.MULTIPLE_CH) res = self.LM.loglikelihood(self.MULTIPLE_CH)
_RES, _res = [r[0] for r in self.MULTIPLE_CH_RES], [r[0] for r in res] _RES, _res = [r[0] for r in self.MULTIPLE_CH_RES], [r[0] for r in res]
assert np.allclose(_res[0], _RES[0], atol=1e-3) assert np.allclose(_res, _RES, atol=1e-3)
def test_greedy_until(self) -> None: def test_greedy_until(self) -> None:
res = self.LM.greedy_until(self.GREEDY_UNTIL) res = self.LM.greedy_until(self.GREEDY_UNTIL)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment