liblaf-cherries 0.0.12__py3-none-any.whl → 0.0.14__py3-none-any.whl

liblaf/cherries/__init__.pyi

@@ -8,13 +8,13 @@ from .pathutils import (
     config,
     data,
     entrypoint,
-    exp_dir,
     git_root,
     git_root_safe,
     inputs,
     outputs,
     params,
     path,
+    run_dir,
     src,
 )
 from .plugin import (
@@ -61,7 +61,6 @@ __all__ = [
     "data",
     "end",
     "entrypoint",
-    "exp_dir",
     "git_root",
     "git_root_safe",
     "info",
@@ -82,6 +81,7 @@ __all__ = [
     "plugin",
     "presets",
     "run",
+    "run_dir",
     "set_tag",
     "src",
     "start",

liblaf/cherries/_run.py

@@ -17,9 +17,9 @@ def run[C: pydantic.BaseModel, T](main: Callable[[C], T]) -> T:
         ret: T = main(cfg)
     except BaseException as e:
         if isinstance(e, KeyboardInterrupt):
-            run.end("KILLED")
+            run.end(plugin.RunStatus.KILLED)
             raise
-        run.end("FAILED")
+        run.end(plugin.RunStatus.FAILED)
         raise
     else:
         run.end()
@@ -31,7 +31,7 @@ def start() -> plugin.Run:
     run.start()
     run.log_src(_path.entrypoint(absolute=True))
     run.set_tag("cherries.entrypoint", _path.entrypoint(absolute=False))
-    run.set_tag("cherries.exp-dir", _path.exp_dir(absolute=False))
+    run.set_tag("cherries.run-dir", _path.run_dir(absolute=False))
     return plugin.run
 
 

liblaf/cherries/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.12'
-__version_tuple__ = version_tuple = (0, 0, 12)
+__version__ = version = '0.0.14'
+__version_tuple__ = version_tuple = (0, 0, 14)

liblaf/cherries/info/__init__.pyi

@@ -1,5 +1,5 @@
-from ._exp_name import exp_name
 from ._git import git_auto_commit, git_branch, git_commit_sha, git_commit_url, git_info
+from ._name import exp_name, run_name
 
 __all__ = [
     "exp_name",
@@ -8,4 +8,5 @@ __all__ = [
     "git_commit_sha",
     "git_commit_url",
     "git_info",
+    "run_name",
 ]

liblaf/cherries/info/{_exp_name.py → _name.py}

@@ -1,7 +1,10 @@
+from pathlib import Path
+
 import git.exc
 from environs import env
 
 from liblaf import grapes
+from liblaf.cherries import pathutils as _path
 
 from ._git import git_info
 
@@ -17,3 +20,11 @@ def exp_name() -> str:
         return "Default"
     else:
         return info.repo
+
+
+def run_name() -> str:
+    run_dir: Path = _path.run_dir(absolute=False)
+    run_name: str = _path.as_posix(run_dir)
+    run_name = run_name.removeprefix("exp")
+    run_name = run_name.removeprefix("/")
+    return run_name

liblaf/cherries/pathutils/__init__.pyi

@@ -1,5 +1,5 @@
 from ._convert import as_os_path, as_path, as_posix
-from ._path import entrypoint, exp_dir, git_root, git_root_safe
+from ._path import entrypoint, git_root, git_root_safe, run_dir
 from ._special import config, data, inputs, outputs, params, path, src
 
 __all__ = [
@@ -9,12 +9,12 @@ __all__ = [
     "config",
     "data",
     "entrypoint",
-    "exp_dir",
     "git_root",
     "git_root_safe",
     "inputs",
     "outputs",
     "params",
     "path",
+    "run_dir",
     "src",
 ]

liblaf/cherries/pathutils/_path.py

@@ -15,13 +15,6 @@ def entrypoint(*, absolute: bool = False) -> Path:
     return _entrypoint_relative()
 
 
-@utils.cache
-def exp_dir(*, absolute: bool = False) -> Path:
-    if absolute:
-        return _exp_dir_absolute()
-    return _exp_dir_relative()
-
-
 @utils.cache
 def git_root() -> Path:
     entrypoint: Path = _entrypoint_absolute()
@@ -38,6 +31,13 @@ def git_root_safe() -> Path:
         return _entrypoint_absolute().parent
 
 
+@utils.cache
+def run_dir(*, absolute: bool = False) -> Path:
+    if absolute:
+        return _run_dir_absolute()
+    return _run_dir_relative()
+
+
 @utils.cache
 def _entrypoint_absolute() -> Path:
     path = Path(sys.argv[0])
@@ -51,7 +51,7 @@ def _entrypoint_relative() -> Path:
 
 
 @utils.cache
-def _exp_dir_absolute() -> Path:
+def _run_dir_absolute() -> Path:
     entrypoint: Path = _entrypoint_absolute()
     for path in entrypoint.parents:
         if (path / "exp.cherries.toml").is_file():
@@ -62,7 +62,7 @@ def _exp_dir_absolute() -> Path:
 
 
 @utils.cache
-def _exp_dir_relative() -> Path:
-    absolute: Path = _exp_dir_absolute()
+def _run_dir_relative() -> Path:
+    absolute: Path = _run_dir_absolute()
     root: Path = git_root_safe()
     return absolute.relative_to(root)

liblaf/cherries/pathutils/_special.py

@@ -2,7 +2,7 @@ from pathlib import Path
 
 from liblaf.grapes.typed import PathLike
 
-from ._path import exp_dir
+from ._path import run_dir
 
 
 def config(
@@ -44,7 +44,7 @@ def src(path: PathLike = "", *, mkdir: bool = True, prefix: PathLike = "src") ->
 def _path(path: PathLike = "", *, mkdir: bool = True, prefix: PathLike = "") -> Path:
     path = Path(path)
     if not path.is_absolute():
-        path = exp_dir(absolute=True) / prefix / path
+        path = run_dir(absolute=True) / prefix / path
     if mkdir:
         path.parent.mkdir(parents=True, exist_ok=True)
     return path

liblaf/cherries/plugin/_dvc.py

@@ -3,6 +3,8 @@ from pathlib import Path
 from typing import override
 
 import attrs
+import git
+import git.exc
 
 from liblaf.cherries import pathutils as _path
 from liblaf.cherries.typed import PathLike
@@ -25,7 +27,9 @@ class DvcLogArtifact(LogArtifact):
         self, local_path: PathLike, artifact_path: PathLike | None = None, **kwargs
     ) -> Path:
         local_path: Path = _path.as_path(local_path)
-        sp.run(["dvc", "add", local_path], check=False)
+        if check_ignore(local_path) or tracked_by_git(local_path):
+            return local_path
+        sp.run(["dvc", "add", local_path], check=True)
         return local_path
 
 
@@ -36,7 +40,9 @@ class DvcLogArtifacts(LogArtifacts):
         self, local_dir: PathLike, artifact_path: PathLike | None = None, **kwargs
     ) -> Path:
         local_dir: Path = _path.as_path(local_dir)
-        sp.run(["dvc", "add", local_dir], check=False)
+        if check_ignore(local_dir) or tracked_by_git(local_dir):
+            return local_dir
+        sp.run(["dvc", "add", local_dir], check=True)
         return local_dir
 
 
@@ -45,3 +51,13 @@ def check_ignore(local_path: PathLike) -> bool:
         ["dvc", "check-ignore", local_path], check=False
     )
     return proc.returncode == 0
+
+
+def tracked_by_git(local_path: PathLike) -> bool:
+    local_path: Path = _path.as_path(local_path).absolute()
+    try:
+        repo = git.Repo(search_parent_directories=True)
+        repo.git.ls_files(local_path, error_unmatch=True)
+    except git.exc.GitCommandError:
+        return False
+    return True

liblaf/cherries/plugin/_logging.py

@@ -25,7 +25,7 @@ class LoggingStart(Start):
         grapes.init_logging(
             handlers=[
                 grapes.logging.rich_handler(),
-                grapes.logging.file_handler(file=run.exp_dir / "run.log"),
-                grapes.logging.jsonl_handler(run.exp_dir / "run.log.jsonl"),
+                grapes.logging.file_handler(sink=run.exp_dir / "run.log"),
+                grapes.logging.jsonl_handler(sink=run.exp_dir / "run.log.jsonl"),
             ]
         )

liblaf/cherries/plugin/_mlflow.py

@@ -3,7 +3,6 @@ from typing import Any, override
 
 import attrs
 import mlflow
-import mlflow.entities
 
 from liblaf.cherries import info as _info
 from liblaf.cherries import pathutils as _path
@@ -34,6 +33,9 @@ class MlflowLogArtifact(LogArtifact):
     def __call__(
         self, local_path: PathLike, artifact_path: PathLike | None = None, **kwargs
     ) -> Path:
+        local_path: Path = _path.as_path(local_path)
+        if (dvc_file := local_path.with_name(local_path.name + ".dvc")).is_file():
+            local_path = dvc_file
         mlflow.log_artifact(
             _path.as_os_path(local_path), _path.as_posix(artifact_path), **kwargs
         )
@@ -46,6 +48,9 @@ class MlflowLogArtifacts(LogArtifacts):
     def __call__(
         self, local_dir: PathLike, artifact_path: PathLike | None = None, **kwargs
     ) -> Path:
+        local_dir: Path = _path.as_path(local_dir)
+        if (dvc_file := local_dir.with_name(local_dir.name + ".dvc")).is_file():
+            local_dir = dvc_file
         mlflow.log_artifact(
             _path.as_os_path(local_dir), _path.as_posix(artifact_path), **kwargs
         )
@@ -80,4 +85,4 @@ class MlflowStart(Start):
     @override
     def __call__(self) -> None:
         mlflow.set_experiment(_info.exp_name())
-        mlflow.start_run()
+        mlflow.start_run(run_name=_info.run_name())

liblaf/cherries/plugin/_run.py

@@ -27,7 +27,7 @@ class Run:
 
     @functools.cached_property
     def exp_dir(self) -> Path:
-        return _path.exp_dir(absolute=True)
+        return _path.run_dir(absolute=True)
 
     @property
     def exp_id(self) -> str:

{liblaf_cherries-0.0.12.dist-info → liblaf_cherries-0.0.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: liblaf-cherries
-Version: 0.0.12
+Version: 0.0.14
 Summary: Add your description here
 Project-URL: Changelog, https://github.com/liblaf/cherries/blob/main/CHANGELOG.md
 Project-URL: Documentation, https://liblaf.github.io/cherries/
@@ -35,7 +35,7 @@ Requires-Dist: dvc[webdav]<4,>=3.59.2
 Requires-Dist: environs<15,>=14.1.1
 Requires-Dist: gitpython<4,>=3.1.44
 Requires-Dist: lazy-loader<0.5,>=0.4
-Requires-Dist: liblaf-grapes<0.2,>=0.1.21
+Requires-Dist: liblaf-grapes<0.2,>=0.1.25
 Requires-Dist: loguru<0.8,>=0.7.3
 Requires-Dist: mlflow<3,>=2.22.0
 Requires-Dist: pydantic-settings<3,>=2.9.1

{liblaf_cherries-0.0.12.dist-info → liblaf_cherries-0.0.14.dist-info}/RECORD

@@ -1,35 +1,35 @@
 liblaf/cherries/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/__init__.pyi,sha256=SuZgWY0gRNMTU-rAapdJ5mCHyYqeQmlR9DosaViUx1A,1370
+liblaf/cherries/__init__.pyi,sha256=ShYXNS05TvRjr5XTKurtBiToTR3T1JX2Ub0njjbMsuM,1370
 liblaf/cherries/_config.py,sha256=rm-Y6roi8hBvQYdH-VQh_ovWCyVsX_7R0x1jokBPCDM,741
-liblaf/cherries/_run.py,sha256=aJonztFyqBeg9otqMGwkh2AOEjtFh85sC1riBa1ty38,1055
-liblaf/cherries/_version.py,sha256=yXPCLHqHWroxl6PYku74xz7jqX7Xrf288eRJfEDWwBo,513
+liblaf/cherries/_run.py,sha256=vbJ5-Fh8azNhemVf4qyNUc5RaFA8HgFHZOmzdRMPF2k,1085
+liblaf/cherries/_version.py,sha256=JAP571q66KmOqXyFj53ns1X0_trh2aTxt2bW41ljMkc,513
 liblaf/cherries/_version.pyi,sha256=Pnv4Bxw13LHeuVkPLPsTtnp4N4jOGcAfFJw05uMMgBY,108
 liblaf/cherries/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 liblaf/cherries/typed.py,sha256=mim8QVtwczTSHyw5mhEdfFcXis9o32n0CZyu8BrEorE,50
 liblaf/cherries/info/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/info/__init__.pyi,sha256=UsHH6XpE_RRre9sJdwgzOnV9HElnE00ltRKnfXOQso4,252
-liblaf/cherries/info/_exp_name.py,sha256=dlC1OIg9RM2oYW7qsWCF6AGV0z6bbQxpJOsdft-KUyk,454
+liblaf/cherries/info/__init__.pyi,sha256=beKfSRY1wVsTPBOkWZ7ukmMZo0NKHzp00Q51cFUArbM,274
 liblaf/cherries/info/_git.py,sha256=oy1xx23sXWA7RqQuLGc3HxcLCF7zHTSpZvZ9pukJGdI,1232
+liblaf/cherries/info/_name.py,sha256=RhB0tqygzBNTS6ZK9unLu7DeuiQaudAvIyeHj0nf2HA,752
 liblaf/cherries/pathutils/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/pathutils/__init__.pyi,sha256=mr3Dk8ragpNGXmXJriMt21vc7YmqUD6X587exFe9tCI,413
+liblaf/cherries/pathutils/__init__.pyi,sha256=JEotcjG2QNC3LaVM7aoLd7l0Ny6m0QyQKeePefS22LM,413
 liblaf/cherries/pathutils/_convert.py,sha256=JTO9vETXvj7f4GTtIbOmAoDNEDc_4hoEHZ7ujbyTgS8,859
-liblaf/cherries/pathutils/_path.py,sha256=igdLrCqk6aIX36V8c1LClhrFViDFWx4WbiR67BcbVIY,1548
-liblaf/cherries/pathutils/_special.py,sha256=b5D3PtW__u0Zpiv69OYoV7TTYc6Dgs4Bu-P8jwrROV4,1415
+liblaf/cherries/pathutils/_path.py,sha256=35WgGghvXq56mFCqT7jYaPo7mO3rnRyRdBh6--woBxE,1548
+liblaf/cherries/pathutils/_special.py,sha256=UZ9dAPj7EAqlRqzQD_b_Z-uCHjv6W_gtMs1Wfr_nDuc,1415
 liblaf/cherries/plugin/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
 liblaf/cherries/plugin/__init__.pyi,sha256=2PPdTp-vfgP34sKGnjCR1oMG5ljTmkZkabr_LV74tXA,1299
 liblaf/cherries/plugin/_abc.py,sha256=H-NH6Gpv3dTJKjaekuAKSn_H0TJPxF-TLa5UU51NNjg,3748
-liblaf/cherries/plugin/_dvc.py,sha256=xmMhgZ8QTT-G15_BfpGzYqHjAa7NvspUkuKJvpIinfs,1277
+liblaf/cherries/plugin/_dvc.py,sha256=1cMcKJmO-9FFWjiQ_V-BqmF9XaKmOiwdE9y3aeFoodY,1800
 liblaf/cherries/plugin/_git.py,sha256=AY6AugwyqtcCAaX475zDINFHgHG1UKupWplquEPbjWU,1396
-liblaf/cherries/plugin/_logging.py,sha256=6ZlPQMSv1QmovjOPL6VMmJJOyufU1KCN1W7lOvkGcZs,779
-liblaf/cherries/plugin/_mlflow.py,sha256=1AV_jIUgs2EYZU-7_e0EtkgYN3cBPS2Yjrs5AUu5v1Y,1950
-liblaf/cherries/plugin/_run.py,sha256=XqMOEFDqba8xOaUIynJwS1Dg6JGVpqtWsHG_XJenypk,3287
+liblaf/cherries/plugin/_logging.py,sha256=EJ-6VgY8Fyn_o2R68diqT7vwnOnuJBPQU7IjwvjKVco,784
+liblaf/cherries/plugin/_mlflow.py,sha256=JS0oGojDyPTUF-Q8ZUJrO36hUXckNFeViAIjJ6lo3ng,2287
+liblaf/cherries/plugin/_run.py,sha256=ldEzxD-PrAiHBwMGjQKiU9eyc0sriwKXPEU6JbId6kE,3287
 liblaf/cherries/presets/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
 liblaf/cherries/presets/__init__.pyi,sha256=ka0zjVDb1kWn1VuDGjAzMX5S9sVKKYcxKelp4EwUBf8,53
 liblaf/cherries/presets/_default.py,sha256=hhe10JzMui8WJhvoLlOfDCHMRlPmdnzRSduFH7rCtP0,742
 liblaf/cherries/utils/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
 liblaf/cherries/utils/__init__.pyi,sha256=F5aTcXpWVmUoctPbLfmQXKyuXYRspAIjaIzfL1_3Lrw,51
 liblaf/cherries/utils/_functools.py,sha256=0Puwvj1Wq4kp3S--hI-CXwUBZ56AtfkqIzFHllQtuug,181
-liblaf_cherries-0.0.12.dist-info/METADATA,sha256=EznK6Za_uBu_9awxVrZzzkpR2DVzMZdpIWry4H5x5RY,1944
-liblaf_cherries-0.0.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-liblaf_cherries-0.0.12.dist-info/licenses/LICENSE,sha256=Ph4NzyU3lGVDeYv-mf8aRmImH8v9rVL9F362FV4G6Ow,1063
-liblaf_cherries-0.0.12.dist-info/RECORD,,
+liblaf_cherries-0.0.14.dist-info/METADATA,sha256=_s8gjrpYQBGHkxYz_D9ibagwI478PgQujoy1AI2vnD0,1944
+liblaf_cherries-0.0.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+liblaf_cherries-0.0.14.dist-info/licenses/LICENSE,sha256=Ph4NzyU3lGVDeYv-mf8aRmImH8v9rVL9F362FV4G6Ow,1063
+liblaf_cherries-0.0.14.dist-info/RECORD,,