Fix lambda value assert in NormalTS

m-wojnar · Apr 13, 2024 · 753f879 · 753f879
1 parent c02eb5b
commit 753f879
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/reinforced_lib/agents/mab/normal_thompson_sampling.py b/reinforced_lib/agents/mab/normal_thompson_sampling.py
@@ -67,7 +67,7 @@ def __init__(
     ) -> None:
         assert alpha > 0
         assert beta > 0
-        assert lam > 0
+        assert lam >= 0
 
         self.n_arms = n_arms