fix(reforcexy): ensure policy_kwargs dict is initialized

Signed-off-by: Jérôme Benoit <jerome.benoit@sap.com>

diff --git a/ReforceXY/user_data/freqaimodels/ReforceXY.py b/ReforceXY/user_data/freqaimodels/ReforceXY.py
index c180825de93924b655897137424c2ed7e3abe79c..456fbcf5e60bcb4977525bd62c0c90fa5e5f803f 100644 (file)
--- a/ReforceXY/user_data/freqaimodels/ReforceXY.py
+++ b/ReforceXY/user_data/freqaimodels/ReforceXY.py
@@ -231,6 +231,8 @@ class ReforceXY(BaseReinforcementLearningModel):
             logger.info("Clip range linear schedule enabled, initial value: %s", _cr)
 
         net_arch = model_params.get("policy_kwargs", {}).get("net_arch", [128, 128])
+        if not model_params.get("policy_kwargs"):
+            model_params["policy_kwargs"] = {}
         model_params["policy_kwargs"].update(
             {
                 "net_arch": net_arch,