Fixing formatting of modified files

Farama-Foundation · Jun 24, 2024 · 8b992cc · 8b992cc
1 parent fc23ffc
commit 8b992cc
Show file tree

Hide file tree

Showing 7 changed files with 45 additions and 13 deletions.
diff --git a/pettingzoo/mpe/_mpe_utils/simple_env.py b/pettingzoo/mpe/_mpe_utils/simple_env.py
@@ -122,7 +122,7 @@ def __init__(
         # This will be used to scale the rendering
         all_poses = [entity.state.p_pos for entity in self.world.entities]
         self.original_cam_range = np.max(np.abs(np.array(all_poses)))
-        
+
         self.steps = 0
 
         self.current_actions = [None] * self.num_agents
@@ -327,12 +327,8 @@ def draw(self):
             else:
                 radius = entity.size * 350
 
-            pygame.draw.circle(
-                self.screen, entity.color * 200, (x, y), radius
-            )  
-            pygame.draw.circle(
-                self.screen, (0, 0, 0), (x, y), radius, 1
-            )  # borders
+            pygame.draw.circle(self.screen, entity.color * 200, (x, y), radius)
+            pygame.draw.circle(self.screen, (0, 0, 0), (x, y), radius, 1)  # borders
             assert (
                 0 < x < self.width and 0 < y < self.height
             ), f"Coordinates {(x, y)} are out of bounds."

diff --git a/pettingzoo/mpe/simple/simple.py b/pettingzoo/mpe/simple/simple.py
@@ -54,7 +54,13 @@
 
 
 class raw_env(SimpleEnv, EzPickle):
-    def __init__(self, max_cycles=25, continuous_actions=False, render_mode=None, dynamic_rescaling=False):
+    def __init__(
+        self,
+        max_cycles=25,
+        continuous_actions=False,
+        render_mode=None,
+        dynamic_rescaling=False,
+    ):
         EzPickle.__init__(
             self,
             max_cycles=max_cycles,

diff --git a/pettingzoo/mpe/simple_adversary/simple_adversary.py b/pettingzoo/mpe/simple_adversary/simple_adversary.py
@@ -64,7 +64,14 @@
 
 
 class raw_env(SimpleEnv, EzPickle):
-    def __init__(self, N=2, max_cycles=25, continuous_actions=False, render_mode=None, dynamic_rescaling=False):
+    def __init__(
+        self,
+        N=2,
+        max_cycles=25,
+        continuous_actions=False,
+        render_mode=None,
+        dynamic_rescaling=False,
+    ):
         EzPickle.__init__(
             self,
             N=N,

diff --git a/pettingzoo/mpe/simple_crypto/simple_crypto.py b/pettingzoo/mpe/simple_crypto/simple_crypto.py
@@ -75,7 +75,13 @@
 
 
 class raw_env(SimpleEnv, EzPickle):
-    def __init__(self, max_cycles=25, continuous_actions=False, render_mode=None, dynamic_rescaling=False):
+    def __init__(
+        self,
+        max_cycles=25,
+        continuous_actions=False,
+        render_mode=None,
+        dynamic_rescaling=False,
+    ):
         EzPickle.__init__(
             self,
             max_cycles=max_cycles,

diff --git a/pettingzoo/mpe/simple_push/simple_push.py b/pettingzoo/mpe/simple_push/simple_push.py
@@ -60,7 +60,13 @@
 
 
 class raw_env(SimpleEnv, EzPickle):
-    def __init__(self, max_cycles=25, continuous_actions=False, render_mode=None, dynamic_rescaling=False):
+    def __init__(
+        self,
+        max_cycles=25,
+        continuous_actions=False,
+        render_mode=None,
+        dynamic_rescaling=False,
+    ):
         EzPickle.__init__(
             self,
             max_cycles=max_cycles,

diff --git a/pettingzoo/mpe/simple_reference/simple_reference.py b/pettingzoo/mpe/simple_reference/simple_reference.py
@@ -66,7 +66,12 @@
 
 class raw_env(SimpleEnv, EzPickle):
     def __init__(
-        self, local_ratio=0.5, max_cycles=25, continuous_actions=False, render_mode=None, dynamic_rescaling=False
+        self,
+        local_ratio=0.5,
+        max_cycles=25,
+        continuous_actions=False,
+        render_mode=None,
+        dynamic_rescaling=False,
     ):
         EzPickle.__init__(
             self,

diff --git a/pettingzoo/mpe/simple_speaker_listener/simple_speaker_listener.py b/pettingzoo/mpe/simple_speaker_listener/simple_speaker_listener.py
@@ -60,7 +60,13 @@
 
 
 class raw_env(SimpleEnv, EzPickle):
-    def __init__(self, max_cycles=25, continuous_actions=False, render_mode=None, dynamic_rescaling=False):
+    def __init__(
+        self,
+        max_cycles=25,
+        continuous_actions=False,
+        render_mode=None,
+        dynamic_rescaling=False,
+    ):
         EzPickle.__init__(
             self,
             max_cycles=max_cycles,