saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-30 20:18:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update tests

Browse source
This commit is contained in:
Antonin Raffin 2020-02-11 16:42:25 +01:00
parent b7dcc8d58e
commit 2afcf395b9
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/test_save_load.py
View file

@ -137,10 +137,9 @@ def test_exclude_include_saved_params(model_class):
@pytest.mark.parametrize("model_class", [SAC, TD3])
def test_save_load_replay_buffer(model_class):
log_folder = 'logs'
replay_path = os.path.join('logs', 'replay_buffer.pkl')
replay_path = os.path.join(log_folder, 'replay_buffer.pkl')
os.makedirs(log_folder, exist_ok=True)
buffer_size = 1000
model = model_class('MlpPolicy', 'Pendulum-v0', buffer_size=buffer_size)
model = model_class('MlpPolicy', 'Pendulum-v0', buffer_size=1000)
model.learn(500)
old_replay_buffer = deepcopy(model.replay_buffer)
model.save_replay_buffer(log_folder)