pre-PR lint

Author: axel-tanner

mujoco_playground/_src/locomotion/t1/randomize.py

@@ -18,7 +18,6 @@
 from mujoco import mjx
 import numpy as np
 
-
 FLOOR_GEOM_ID = 0
 TORSO_BODY_ID = 1
 ANKLE_JOINT_IDS = np.array([[21, 22, 27, 28]])
@@ -30,7 +29,7 @@ def rand_dynamics(rng):
     # Floor friction: =U(0.4, 1.0).
     rng, key = jax.random.split(rng)
     geom_friction = model.geom_friction.at[FLOOR_GEOM_ID, 0].set(
-        jax.random.uniform(key, minval=0.2, maxval=.6)
+        jax.random.uniform(key, minval=0.2, maxval=0.6)
     )
 
     rng, key = jax.random.split(rng)

mujoco_playground/_src/registry.py

@@ -31,9 +31,7 @@
 
 # A tuple containing all available environment names across all suites.
 ALL_ENVS = (
-    dm_control_suite.ALL_ENVS
-    + locomotion.ALL_ENVS
-    + manipulation.ALL_ENVS
+    dm_control_suite.ALL_ENVS + locomotion.ALL_ENVS + manipulation.ALL_ENVS
 )
 
 

mujoco_playground/config/locomotion_params.py

@@ -88,8 +88,8 @@ def brax_ppo_config(env_name: str) -> config_dict.ConfigDict:
         policy_obs_key="state",
         value_obs_key="privileged_state",
     )
-  
-  elif env_name in ("Go2JoystickFlatTerrain","Go2JoystickRoughTerrain"):
+
+  elif env_name in ("Go2JoystickFlatTerrain", "Go2JoystickRoughTerrain"):
     rl_config.num_timesteps = 200_000_000
     rl_config.num_evals = 10
     rl_config.num_resets_per_eval = 1