DummyVecEnv.render() no longer returns the expected state after [this](…

…DLR-RM/stable-baselines3#1327) pull request got merged in `stable-baselines3`. This commit fixes AI4Finance-Foundation#1010
AkashKarnatak · Jun 10, 2023 · 311b1f4 · 311b1f4
1 parent c505d42
commit 311b1f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/finrl/agents/stablebaselines3/models.py b/finrl/agents/stablebaselines3/models.py
@@ -316,7 +316,7 @@ def DRL_prediction(
             trade_obs, rewards, dones, info = trade_env.step(action)
             if i == (len(trade_data.index.unique()) - 2):
                 # print(env_test.render())
-                last_state = trade_env.render()
+                last_state = trade_env.envs[0].render()
 
         df_last_state = pd.DataFrame({"last_state": last_state})
         df_last_state.to_csv(f"results/last_state_{name}_{i}.csv", index=False)