Merge branch 'dev' of github.com:kalininalab/SMTB2024 into dev

kalininalab · Aug 16, 2024 · dcf30f3 · dcf30f3
2 parents 9167693 + 3a44a68
commit dcf30f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/smtb/train.py b/smtb/train.py
@@ -1,11 +1,11 @@
 import argparse
 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