pre-commit formatting

Acellera · May 27, 2024 · 45d21d1 · 45d21d1
1 parent 9b8e10b
commit 45d21d1
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/acegen/data/chem_utils.py b/acegen/data/chem_utils.py
@@ -1,4 +1,5 @@
 import warnings
+
 import numpy as np
 
 from rdkit.Chem import AllChem as Chem, Draw

diff --git a/acegen/data/smiles_dataset.py b/acegen/data/smiles_dataset.py
@@ -1,6 +1,6 @@
+import gzip
 import logging
 import os
-import gzip
 from pathlib import Path
 
 import numpy as np

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: