From c5d3d1c65718919c50983edb5fb2b839da46411b Mon Sep 17 00:00:00 2001 From: Laurent Date: Tue, 15 Oct 2024 13:19:05 +0000 Subject: [PATCH] clean up wandb import, remove deprecated warnings handling --- src/refiners/training_utils/wandb.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/src/refiners/training_utils/wandb.py b/src/refiners/training_utils/wandb.py index 035dd93..539431b 100644 --- a/src/refiners/training_utils/wandb.py +++ b/src/refiners/training_utils/wandb.py @@ -1,19 +1,14 @@ -import warnings from abc import ABC from pathlib import Path from typing import Any, Literal +import wandb from PIL import Image from refiners.training_utils.callback import Callback, CallbackConfig from refiners.training_utils.config import BaseConfig from refiners.training_utils.trainer import Trainer, register_callback -with warnings.catch_warnings(): - # TODO: remove when https://github.com/wandb/wandb/issues/6711 gets solved - warnings.filterwarnings("ignore", category=DeprecationWarning, message="pkg_resources is deprecated as an API") - import wandb - number = float | int WandbLoggable = number | Image.Image | list[number] | dict[str, list[number]]