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

(training_utils) neptune callback pydantic fix

Browse source
This commit is contained in:
Laurent 2024-05-28 16:56:07 +00:00 committed by Laureηt
parent 3ad7f592db
commit 7dba1c8034
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/refiners/training_utils/neptune.py
View file

@ -1,5 +1,5 @@
from abc import ABC from abc import ABC
from os import PathLike from pathlib import Path
from typing import Any, Literal from typing import Any, Literal
from neptune import Run, init_run # type: ignore from neptune import Run, init_run # type: ignore
@ -43,7 +43,7 @@ class NeptuneConfig(CallbackConfig):
proxies: dict[str, str] | None = None proxies: dict[str, str] | None = None
capture_traceback: bool = True capture_traceback: bool = True
git_ref: GitRef | GitRefDisabled | None = None git_ref: GitRef | GitRefDisabled | None = None
dependencies: PathLike[str] | str | None = None dependencies: Path | str | None = None
async_lag_callback: NeptuneObjectCallback | None = None async_lag_callback: NeptuneObjectCallback | None = None
async_no_progress_callback: NeptuneObjectCallback | None = None async_no_progress_callback: NeptuneObjectCallback | None = None
async_lag_threshold: float = ASYNC_LAG_THRESHOLD async_lag_threshold: float = ASYNC_LAG_THRESHOLD