flake_style_format

dizoo/gym_soccer/envs/gym_soccer_env.py

@@ -19,7 +19,7 @@ class GymSoccerEnv(BaseEnv):
 
     def __init__(self, cfg: dict = {}) -> None:
         self._cfg = cfg
-        self._act_scale = cfg.get('act_scale',True)
+        self._act_scale = cfg.get('act_scale', True)
         self._env_id = cfg.env_id
         assert self._env_id in self.default_env_id
         self._init_flag = False
@@ -36,11 +36,11 @@ class GymSoccerEnv(BaseEnv):
 
     def step(self, action: List) -> BaseEnvTimestep:
         if self._act_scale:
-            action[1][0] = affine_transform(action[1][0],min_val=0,max_val=100)
-            action[2][0] = affine_transform(action[2][0],min_val=-180,max_val=180)
-            action[3][0] = affine_transform(action[3][0],min_val=-180,max_val=180)
-            action[4][0] = affine_transform(action[4][0],min_val=0,max_val=100)
-            action[5][0] = affine_transform(action[5][0],min_val=-180,max_val=180)
+            action[1][0] = affine_transform(action[1][0], min_val=0, max_val=100)
+            action[2][0] = affine_transform(action[2][0], min_val=-180, max_val=180)
+            action[3][0] = affine_transform(action[3][0], min_val=-180, max_val=180)
+            action[4][0] = affine_transform(action[4][0], min_val=0, max_val=100)
+            action[5][0] = affine_transform(action[5][0], min_val=-180, max_val=180)
 
         obs, rew, done, info = self._env.step(action)
         self._final_eval_reward += rew