From 9ce21f569ecac03f15f2ad669fde3dd4a512f8cc Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Mon, 15 Apr 2024 21:47:54 +0200 Subject: Format --- training/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'training/utils.py') diff --git a/training/utils.py b/training/utils.py index d1801a7..f0a0d4d 100644 --- a/training/utils.py +++ b/training/utils.py @@ -1,6 +1,6 @@ """Util functions for training with hydra and pytorch lightning.""" -from typing import List, Type import warnings +from typing import List, Type import hydra from loguru import logger as log @@ -14,6 +14,7 @@ from pytorch_lightning.loggers import Logger from pytorch_lightning.loggers.wandb import WandbLogger from pytorch_lightning.utilities import rank_zero_only from tqdm import tqdm + import wandb -- cgit v1.2.3-70-g09d2