From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Mon, 3 Mar 2025 21:13:48 +0000 (+0100)
Subject: fix(reforcexy): parse properly policy_kwargs tunable section
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=91732a7d9c8625228035ac612349949519cab658;p=freqai-strategies.git

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 c485ed0..fa53dbf 100644
--- 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