From 753f8790dcd9ea7217bfdc14c4920718ec5db1a5 Mon Sep 17 00:00:00 2001 From: Maksymilian Wojnar Date: Sat, 13 Apr 2024 19:13:33 +0200 Subject: [PATCH] Fix lambda value assert in NormalTS --- reinforced_lib/agents/mab/normal_thompson_sampling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/reinforced_lib/agents/mab/normal_thompson_sampling.py b/reinforced_lib/agents/mab/normal_thompson_sampling.py index 9d661a0..d6cd9ba 100644 --- 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