Merge branch 'main' of https://github.com/online-ml/river

kulbachcedric · Jun 4, 2023 · 15e9870 · 15e9870
2 parents c335e3b + dedb8dd
commit 15e9870
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/river/bandit/evaluate.py b/river/bandit/evaluate.py
@@ -158,7 +158,7 @@ def evaluate(
  tuple[
  list[ArmID], # arms available to pull from
  ArmID, # arm that was pulled
- float | None, # probability of pulling the arm
+ typing.Union[float, None], # noqa: UP007, probability of pulling the arm
  float, # reward
  ]
 ]