Lint

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()

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