Minor improvements

Digitalized-Energy-Systems · Nov 18, 2024 · 009607e · 009607e
1 parent 4db5808
commit 009607e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/opfgym/reward.py b/opfgym/reward.py
@@ -33,7 +33,7 @@ def prepare_reward_scaling(self,
         reward_scaler = select_reward_scaler(reward_scaling)
         try:
             scaling_params.update(reward_scaler(**scaling_params))
-        except:
+        except TypeError:
             scaling_params = estimate_reward_distribution(env, **scaling_params)
             scaling_params.update(reward_scaler(**scaling_params))