Skip to content

Commit c5d3d1c

Browse files
LaurentLaurent2916
authored andcommitted
clean up wandb import, remove deprecated warnings handling
1 parent 241abfa commit c5d3d1c

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

src/refiners/training_utils/wandb.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,14 @@
1-
import warnings
21
from abc import ABC
32
from pathlib import Path
43
from typing import Any, Literal
54

5+
import wandb
66
from PIL import Image
77

88
from refiners.training_utils.callback import Callback, CallbackConfig
99
from refiners.training_utils.config import BaseConfig
1010
from refiners.training_utils.trainer import Trainer, register_callback
1111

12-
with warnings.catch_warnings():
13-
# TODO: remove when https://github.com/wandb/wandb/issues/6711 gets solved
14-
warnings.filterwarnings("ignore", category=DeprecationWarning, message="pkg_resources is deprecated as an API")
15-
import wandb
16-
1712
number = float | int
1813
WandbLoggable = number | Image.Image | list[number] | dict[str, list[number]]
1914

0 commit comments

Comments
 (0)