OscervaME
/
stable-baselines3-contrib-sacd
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Lint

This commit is contained in:
Antonin RAFFIN 2020-10-12 20:25:11 +02:00
parent 5033b192cb
commit afe7b132e4
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_run.py
View File

@ -25,9 +25,15 @@ def test_n_critics(n_critics):
) )
model.learn(total_timesteps=500) model.learn(total_timesteps=500)
def test_sde(): def test_sde():
model = TQC( model = TQC(
"MlpPolicy", "Pendulum-v0", policy_kwargs=dict(net_arch=[64], sde_net_arch=[8]), use_sde=True, learning_starts=100, verbose=1 "MlpPolicy",
"Pendulum-v0",
policy_kwargs=dict(net_arch=[64], sde_net_arch=[8]),
use_sde=True,
learning_starts=100,
verbose=1,
) )
model.learn(total_timesteps=500) model.learn(total_timesteps=500)
model.policy.reset_noise() model.policy.reset_noise()

2
tests/test_save_load.py
View File

@ -1,6 +1,5 @@
import os import os
import pathlib import pathlib
import warnings
from collections import OrderedDict from collections import OrderedDict
from copy import deepcopy from copy import deepcopy
@ -11,7 +10,6 @@ import torch as th
from stable_baselines3 import DQN from stable_baselines3 import DQN
from stable_baselines3.common.base_class import BaseAlgorithm from stable_baselines3.common.base_class import BaseAlgorithm
from stable_baselines3.common.identity_env import FakeImageEnv, IdentityEnv, IdentityEnvBox from stable_baselines3.common.identity_env import FakeImageEnv, IdentityEnv, IdentityEnvBox
from stable_baselines3.common.save_util import load_from_pkl, open_path, save_to_pkl
from stable_baselines3.common.utils import get_device from stable_baselines3.common.utils import get_device
from stable_baselines3.common.vec_env import DummyVecEnv from stable_baselines3.common.vec_env import DummyVecEnv