Replace unnecessary SubProcEnv with DummyVecEnv

notadamking · Jul 10, 2019 · d1d19de · d1d19de
1 parent b1ca6f9
commit d1d19de
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/data/params.db b/data/params.db
diff --git a/lib/RLTrader.py b/lib/RLTrader.py
@@ -134,8 +134,8 @@ def optimize_params(self, trial, n_prune_evals_per_trial: int = 2, n_tests_per_e
 
         del test_provider
 
-        train_env = SubprocVecEnv([make_env(train_provider, i) for i in range(1)])
-        validation_env = SubprocVecEnv([make_env(validation_provider, i) for i in range(1)])
+        train_env = DummyVecEnv([lambda: TradingEnv(train_provider)])
+        validation_env = DummyVecEnv([lambda: TradingEnv(validation_provider)])
 
         model_params = self.optimize_agent_params(trial)
         model = self.Model(self.Policy,