From afe7b132e41b59e4314a9e905d105e9910997bd2 Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Mon, 12 Oct 2020 20:25:11 +0200 Subject: [PATCH] Lint --- tests/test_run.py | 8 +++++++- tests/test_save_load.py | 2 -- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/test_run.py b/tests/test_run.py index 241ea92..c2bc941 100644 --- a/tests/test_run.py +++ b/tests/test_run.py @@ -25,9 +25,15 @@ def test_n_critics(n_critics): ) model.learn(total_timesteps=500) + def test_sde(): 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.policy.reset_noise() diff --git a/tests/test_save_load.py b/tests/test_save_load.py index 8a704ac..9a73c77 100644 --- a/tests/test_save_load.py +++ b/tests/test_save_load.py @@ -1,6 +1,5 @@ import os import pathlib -import warnings from collections import OrderedDict from copy import deepcopy @@ -11,7 +10,6 @@ import torch as th from stable_baselines3 import DQN from stable_baselines3.common.base_class import BaseAlgorithm 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.vec_env import DummyVecEnv