File tree Expand file tree Collapse file tree 1 file changed +1
-6
lines changed
src/refiners/training_utils Expand file tree Collapse file tree 1 file changed +1
-6
lines changed Original file line number Diff line number Diff line change 1
- import warnings
2
1
from abc import ABC
3
2
from pathlib import Path
4
3
from typing import Any , Literal
5
4
5
+ import wandb
6
6
from PIL import Image
7
7
8
8
from refiners .training_utils .callback import Callback , CallbackConfig
9
9
from refiners .training_utils .config import BaseConfig
10
10
from refiners .training_utils .trainer import Trainer , register_callback
11
11
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
-
17
12
number = float | int
18
13
WandbLoggable = number | Image .Image | list [number ] | dict [str , list [number ]]
19
14
You can’t perform that action at this time.
0 commit comments