From fc08439ec032eb2aab1a520297a3b736d61e2bdc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89mile=20Royer?= Date: Mon, 22 Jul 2024 16:31:38 +0200 Subject: [PATCH] Add a float conversion to please MyPy --- river/bandit/test_policies.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/river/bandit/test_policies.py b/river/bandit/test_policies.py index 0bb79f49cc..b1d6507ebf 100644 --- a/river/bandit/test_policies.py +++ b/river/bandit/test_policies.py @@ -111,7 +111,7 @@ def test_better_than_random_policy(policy: bandit.base.Policy, env: gym.Env): arm_id = policy.pull(arm_ids) # type: ignore observation, reward, terminated, truncated, info = env.step(arm_id) policy.update(arm_id, reward) - policy_reward += reward + policy_reward += float(reward) random_arm_id = random_policy.pull(arm_ids) # type: ignore ( @@ -122,7 +122,7 @@ def test_better_than_random_policy(policy: bandit.base.Policy, env: gym.Env): info, ) = random_env.step(random_arm_id) random_policy.update(random_arm_id, reward) - random_reward += reward + random_reward += float(reward) n_successes += policy_reward > random_reward