flake8 fix

bgruening · Oct 16, 2024 · bf00d42 · bf00d42
1 parent 988120f
commit bf00d42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/prot_tools/optuna_tool.py b/tools/prot_tools/optuna_tool.py
@@ -39,14 +39,14 @@
     EsmModel,
     EsmTokenizer,
     PreTrainedModel,
+    set_seed,
     T5EncoderModel,
     T5Tokenizer,
     Trainer,
     TrainerCallback,
     TrainerControl,
     TrainerState,
     TrainingArguments,
-    set_seed,
 )
 from transformers.modeling_outputs import SequenceClassifierOutput
 from transformers.models.t5.modeling_t5 import T5Config, T5PreTrainedModel, T5Stack