diff --git a/tools/prot_tools/optuna_tool.py b/tools/prot_tools/optuna_tool.py
index c36640a7d4..365d0973b4 100644
--- a/tools/prot_tools/optuna_tool.py
+++ b/tools/prot_tools/optuna_tool.py
@@ -39,6 +39,7 @@
     EsmModel,
     EsmTokenizer,
     PreTrainedModel,
+    set_seed,
     T5EncoderModel,
     T5Tokenizer,
     Trainer,
@@ -46,7 +47,6 @@
     TrainerControl,
     TrainerState,
     TrainingArguments,
-    set_seed,
 )
 from transformers.modeling_outputs import SequenceClassifierOutput
 from transformers.models.t5.modeling_t5 import T5Config, T5PreTrainedModel, T5Stack