From 7dba1c8034859f1aba6a68c0b625241dc745e2aa Mon Sep 17 00:00:00 2001 From: Laurent Date: Tue, 28 May 2024 16:56:07 +0000 Subject: [PATCH] (training_utils) neptune callback pydantic fix --- src/refiners/training_utils/neptune.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/refiners/training_utils/neptune.py b/src/refiners/training_utils/neptune.py index 018b014..36e5cfd 100644 --- a/src/refiners/training_utils/neptune.py +++ b/src/refiners/training_utils/neptune.py @@ -1,5 +1,5 @@ from abc import ABC -from os import PathLike +from pathlib import Path from typing import Any, Literal from neptune import Run, init_run # type: ignore @@ -43,7 +43,7 @@ class NeptuneConfig(CallbackConfig): proxies: dict[str, str] | None = None capture_traceback: bool = True git_ref: GitRef | GitRefDisabled | None = None - dependencies: PathLike[str] | str | None = None + dependencies: Path | str | None = None async_lag_callback: NeptuneObjectCallback | None = None async_no_progress_callback: NeptuneObjectCallback | None = None async_lag_threshold: float = ASYNC_LAG_THRESHOLD