pre-commit formatting

Acellera · May 30, 2024 · de35e47 · de35e47
1 parent 32b4184
commit de35e47
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/scripts/sac/pretrain_sac.py b/scripts/sac/pretrain_sac.py
@@ -345,9 +345,8 @@ def create_env_fn():
                     "train/reward": episode_rewards.mean().item(),
                     "train/min_reward": episode_rewards.min().item(),
                     "train/max_reward": episode_rewards.max().item(),
-                    "train/episode_length": episode_length.sum().item() / len(
-                        episode_length
-                    ),
+                    "train/episode_length": episode_length.sum().item()
+                    / len(episode_length),
                 }
             )
             if logger:

diff --git a/scripts/sac/sac.py b/scripts/sac/sac.py
@@ -354,9 +354,8 @@ def create_env_fn():
                     "train/reward": episode_rewards.mean().item(),
                     "train/min_reward": episode_rewards.min().item(),
                     "train/max_reward": episode_rewards.max().item(),
-                    "train/episode_length": episode_length.sum().item() / len(
-                        episode_length
-                    ),
+                    "train/episode_length": episode_length.sum().item()
+                    / len(episode_length),
                 }
             )
             if logger: