From 23dcfae1ff62f9bc2ec0a16bbf33a87ba856b6fb Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 30 Sep 2024 00:49:27 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 smtb/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/smtb/train.py b/smtb/train.py
index 103b290..b250bc8 100644
--- a/smtb/train.py
+++ b/smtb/train.py
@@ -1,11 +1,11 @@
 from argparse import Namespace
 from pathlib import Path
 
+import wandb
 from lightning.pytorch import Trainer, seed_everything
 from lightning.pytorch.callbacks import EarlyStopping, ModelCheckpoint, RichProgressBar
 from lightning.pytorch.loggers import WandbLogger
 
-import wandb
 from smtb.data import DownstreamDataModule
 from smtb.model import RegressionModel