diff --git a/tapeagents/finetune/logging_.py b/tapeagents/finetune/logging_.py
index 6333d99e..d2222a12 100644
--- a/tapeagents/finetune/logging_.py
+++ b/tapeagents/finetune/logging_.py
@@ -8,9 +8,8 @@
 
 import datasets
 import transformers
-from omegaconf import DictConfig
-
 import wandb
+from omegaconf import DictConfig
 from wandb.sdk import wandb_run
 
 from .context import accelerator, logger