Skip to content

Commit

Permalink
After [this](DLR-RM/stable-baselines3@9c338f9) by stable-baselines3
Browse files Browse the repository at this point in the history
…, DummyVecEnv.reset() method passes a `seed` argument to its list of envs. Since StockTradingEnv.reset() does not expect any argument it results in an error. This commit fixes AI4Finance-Foundation#1022.
  • Loading branch information
AkashKarnatak committed Jun 10, 2023
1 parent 8853ab1 commit c505d42
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion finrl/meta/env_stock_trading/env_stocktrading.py
Original file line number Diff line number Diff line change
Expand Up @@ -355,7 +355,7 @@ def step(self, actions):

return self.state, self.reward, self.terminal, False, {}

def reset(self):
def reset(self, seed=None):
# initiate state
self.state = self._initiate_state()

Expand Down

0 comments on commit c505d42

Please sign in to comment.