diff --git a/gflownet/policy/mlp.py b/gflownet/policy/mlp.py
index 583760b6..07d6528e 100644
--- 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
index c4f0f397..4eb77ec3 100644
--- 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
index 085622d2..f90d051b 100644
--- 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
index 02cadd71..e078791d 100644
--- 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):