From d60c3a3e37453594b48e1d6618229f63b99a2832 Mon Sep 17 00:00:00 2001
From: AdrianSosic <adrian.sosic@merckgroup.com>
Date: Thu, 5 Sep 2024 21:59:32 +0200
Subject: [PATCH] Explictly mention one-hot encoding requirement

---
 examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py b/examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py
index 455c6ff53..180638a4b 100644
--- a/examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py
+++ b/examples/Multi_Armed_Bandit/bernoulli_multi_armed_bandit.py
@@ -92,7 +92,10 @@ def draw_arm(self, arm_index: int) -> bool:
 n_arms = len(WIN_RATES)
 target = BinaryTarget(name="clicked")
 parameter = CategoricalParameter(
-    name="Bandit Arm", values=[str(i) for i in range(n_arms)]
+    name="Bandit Arm",
+    values=[str(i) for i in range(n_arms)],
+    # For bandits, one-hot parameter encoding is required:
+    encoding="OHE",
 )
 surrogate = BetaBernoulliMultiArmedBanditSurrogate()