From de35e475a9345cff6296faaa6e853b91bb5a505e Mon Sep 17 00:00:00 2001 From: MorganCThomas Date: Thu, 30 May 2024 13:07:50 +0200 Subject: [PATCH] pre-commit formatting --- scripts/sac/pretrain_sac.py | 5 ++--- scripts/sac/sac.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/scripts/sac/pretrain_sac.py b/scripts/sac/pretrain_sac.py index 416fc0ac..535b3557 100644 --- 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 index 48781105..dfcfdf0e 100644 --- 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: