From 4b3701ceb68a4ab485e13bf8303e3af1881ce8e3 Mon Sep 17 00:00:00 2001 From: wassname Date: Sun, 12 Nov 2017 14:16:49 +0800 Subject: [PATCH] fix concat states --- rl_portfolio_management/wrappers/concat_states.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/rl_portfolio_management/wrappers/concat_states.py b/rl_portfolio_management/wrappers/concat_states.py index e8ecf52..adf2109 100644 --- a/rl_portfolio_management/wrappers/concat_states.py +++ b/rl_portfolio_management/wrappers/concat_states.py @@ -38,3 +38,7 @@ class ConcatStates(gym.Wrapper): state = concat_states(state) return state, reward, done, info + + def reset(self): + state = self.env.reset() + return concat_states(state)