further formatting with isort

alexhernandezgarcia · Jun 21, 2024 · 1508e4b · 1508e4b
1 parent d50aeea
commit 1508e4b
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 4 deletions.
diff --git a/gflownet/policy/mlp.py b/gflownet/policy/mlp.py
@@ -1,5 +1,6 @@
-from torch import nn
 from omegaconf import OmegaConf
+from torch import nn
+
 from gflownet.policy.base import Policy
 
 

diff --git a/scripts/crystal/eval_crystalgflownet.py b/scripts/crystal/eval_crystalgflownet.py
@@ -15,6 +15,7 @@
 sys.path.append(str(Path(__file__).resolve().parent.parent))
 
 from crystalrandom import generate_random_crystals
+
 from gflownet.gflownet import GFlowNetAgent
 from gflownet.utils.common import load_gflow_net_from_run_path
 from gflownet.utils.policy import parse_policy_config

diff --git a/scripts/crystal/eval_gflownet.py b/scripts/crystal/eval_gflownet.py
@@ -8,8 +8,8 @@
 from argparse import ArgumentParser
 from pathlib import Path
 
-import pandas as pd
 import numpy as np
+import pandas as pd
 import torch
 from tqdm import tqdm
 

diff --git a/scripts/crystal/sample_uniform_with_rewards.py b/scripts/crystal/sample_uniform_with_rewards.py
@@ -9,11 +9,11 @@
 
 import hydra
 import pandas as pd
+from crystalrandom import generate_random_crystals_uniform
+
 from gflownet.utils.common import chdir_random_subdir
 from gflownet.utils.policy import parse_policy_config
 
-from crystalrandom import generate_random_crystals_uniform
-
 
 @hydra.main(config_path="../../config", config_name="main", version_base="1.1")
 def main(config):