Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-14 09:08:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

clean up wandb import, remove deprecated warnings handling

Browse source
This commit is contained in:
Laurent 2024-10-15 13:19:05 +00:00 committed by Laureηt
parent 241abfafe4
commit c5d3d1c657
1 changed files with 1 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/refiners/training_utils/wandb.py
View file

@ -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]]