Merge pull request #8677 from richardjozsa/develop

Stable baselines 3 seeding update
This commit is contained in:
Matthias
2023-05-25 06:23:34 +02:00
committed by GitHub
2 changed files with 3 additions and 3 deletions

View File

@@ -180,7 +180,7 @@ class BaseEnvironment(gym.Env):
def reset_tensorboard_log(self): def reset_tensorboard_log(self):
self.tensorboard_metrics = {} self.tensorboard_metrics = {}
def reset(self): def reset(self, seed=None):
""" """
Reset is called at the beginning of every episode Reset is called at the beginning of every episode
""" """

View File

@@ -5,7 +5,7 @@
torch==2.0.1 torch==2.0.1
#until these branches will be released we can use this #until these branches will be released we can use this
gymnasium==0.28.1 gymnasium==0.28.1
stable_baselines3==2.0.0a5 stable_baselines3==2.0.0a9
sb3_contrib>=2.0.0a4 sb3_contrib>=2.0.0a9
# Progress bar for stable-baselines3 and sb3-contrib # Progress bar for stable-baselines3 and sb3-contrib
tqdm==4.65.0 tqdm==4.65.0