chore: Refactor code to use boolean comparison instead of string comp…

…arison for is_discrete in EnvStrategyTrain
XingYu-Zhong · Jun 13, 2024 · 5902106 · 5902106
1 parent a715155
commit 5902106
Show file tree

Hide file tree

Showing 8 changed files with 539 additions and 14 deletions.
diff --git a/algomodel/algo_center.py b/algomodel/algo_center.py
@@ -55,8 +55,7 @@ def stable_baselines3_algo(self,algo_model,alog_parameters,env):
             self.vf_coef = float(alog_parameters['vf_coef']) if 'vf_coef' in alog_parameters else 0.5
             self.max_grad_norm = float(
                 alog_parameters['max_grad_norm']) if 'max_grad_norm' in alog_parameters else 0.5
-            alog_parameters['use_sde'] = alog_parameters['use_sde'] == "True"
-            self.use_sde = alog_parameters['use_sde'] if 'use_sde' in alog_parameters else False
+            self.use_sde = alog_parameters.get('use_sde', False) == 'True'
             self.sde_sample_freq = int(
                 alog_parameters['sde_sample_freq']) if 'sde_sample_freq' in alog_parameters else -1
             self.target_kl = alog_parameters['target_kl'] if 'target_kl' in alog_parameters else None