diff --git a/prompting/utils/logging.py b/prompting/utils/logging.py
index 5028b0cb..a95d888f 100644
--- a/prompting/utils/logging.py
+++ b/prompting/utils/logging.py
@@ -5,12 +5,12 @@
 from typing import Any, Literal
 
 import numpy as np
+import wandb
 from loguru import logger
 from pydantic import BaseModel, ConfigDict
 from wandb.wandb_run import Run
 
 import prompting
-import wandb
 from prompting.base.dendrite import DendriteResponseEvent
 from prompting.rewards.reward import WeightedRewardEvent
 from prompting.settings import settings