fix(reforcexy): parse properly policy_kwargs tunable section

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index c485ed0208e5d25f0f889993db32676354bf6a6b..fa53dbf1f8890325339090432b20ba9e13ac513c 100644 (file)
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -242,13 +242,13 @@ class ReforceXY(BaseReinforcementLearningModel):
         else:
             model_params["policy_kwargs"]["net_arch"] = net_arch
 
-        model_params["policy_kwargs"]["activation_fn"] = model_params[
-            "policy_kwargs"
-        ].get("activation_fn", "relu")
+        model_params["policy_kwargs"]["activation_fn"] = get_activation_fn(
+            model_params["policy_kwargs"].get("activation_fn", "relu")
+        )
 
-        model_params["policy_kwargs"]["optimizer_class"] = model_params[
-            "policy_kwargs"
-        ].get("optimizer_class", "adam")
+        model_params["policy_kwargs"]["optimizer_class"] = get_optimizer_class(
+            model_params["policy_kwargs"].get("optimizer_class", "adam")
+        )
 
         return model_params