minor reformat (#2)

* update atari.py

* fix setup.py
pass the pytest

* fix setup.py
pass the pytest

setup.py

@@ -37,11 +37,19 @@ setup(
                                     'examples', 'examples.*',
                                     'docs', 'docs.*']),
     install_requires=[
-        'gym',
+        'gym>=0.15.0',
         'tqdm',
         'numpy',
         'cloudpickle',
         'tensorboard',
         'torch>=1.4.0',
     ],
+    extras_require={
+        'atari': [
+            'atari_py',
+        ],
+        'mujoco': [
+            'mujoco_py',
+        ]
+    },
 )

test/base/test_buffer.py

@@ -1,4 +1,5 @@
 from tianshou.data import ReplayBuffer
+
 if __name__ == '__main__':
     from env import MyTestEnv
 else:  # pytest

test/base/test_collector.py

@@ -11,6 +11,7 @@ else:  # pytest
 
 class MyPolicy(BasePolicy):
     """docstring for MyPolicy"""
+
     def __init__(self):
         super().__init__()
 

tianshou/env/vecenv.py

@@ -1,6 +1,7 @@
 import numpy as np
 from abc import ABC, abstractmethod
 from multiprocessing import Process, Pipe
+
 try:
     import ray
 except ImportError:

tianshou/policy/ddpg.py

@@ -5,6 +5,8 @@ import torch.nn.functional as F
 
 from tianshou.data import Batch
 from tianshou.policy import BasePolicy
+
+
 # from tianshou.exploration import OUNoise