From 41219fe81e1d93b8b2b9ed29c5bab60aaabf2b4d Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 12 Mar 2025 10:16:54 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- lerobot/common/policies/sac/modeling_sac.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lerobot/common/policies/sac/modeling_sac.py b/lerobot/common/policies/sac/modeling_sac.py index 8ea00a1b..afbbc945 100644 --- a/lerobot/common/policies/sac/modeling_sac.py +++ b/lerobot/common/policies/sac/modeling_sac.py @@ -231,7 +231,7 @@ class SACPolicy( # Load or create config if config_file and os.path.exists(config_file): # Load config from file - with open(config_file, "r") as f: + with open(config_file) as f: config_dict = json.load(f) config = SACConfig(**config_dict) else: @@ -690,9 +690,9 @@ class Policy(nn.Module): # Compute standard deviations if self.fixed_std is None: log_std = self.std_layer(outputs) - assert not torch.isnan(log_std).any(), ( - "[ERROR] log_std became NaN after std_layer!" - ) + assert not torch.isnan( + log_std + ).any(), "[ERROR] log_std became NaN after std_layer!" if self.use_tanh_squash: log_std = torch.tanh(log_std)