liblaf-cherries 0.1.6__py3-none-any.whl → 0.2.0__py3-none-any.whl

liblaf/cherries/integration/_exp.py

@@ -1,109 +0,0 @@
-import datetime
-import functools
-from os import PathLike
-from pathlib import Path
-
-import attrs
-import comet_ml as comet
-
-from liblaf.cherries import pathutils as _path
-
-from ._abc import (
-    AddTag,
-    AddTags,
-    End,
-    LogAsset,
-    LogCode,
-    LogMetric,
-    LogMetrics,
-    LogOther,
-    LogOthers,
-    LogParam,
-    LogParams,
-    Start,
-)
-
-
-@attrs.define
-class Experiment:
-    add_tag: AddTag = attrs.field(factory=AddTag, init=False)
-    add_tags: AddTags = attrs.field(factory=AddTags, init=False)
-    end: End = attrs.field(factory=End, init=False)
-    log_asset: LogAsset = attrs.field(factory=LogAsset, init=False)
-    log_code: LogCode = attrs.field(factory=LogCode, init=False)
-    log_metric: LogMetric = attrs.field(factory=LogMetric, init=False)
-    log_metrics: LogMetrics = attrs.field(factory=LogMetrics, init=False)
-    log_other: LogOther = attrs.field(factory=LogOther, init=False)
-    log_others: LogOthers = attrs.field(factory=LogOthers, init=False)
-    log_param: LogParam = attrs.field(factory=LogParam, init=False)
-    log_params: LogParams = attrs.field(factory=LogParams, init=False)
-    start: Start = attrs.field(factory=Start, init=False)
-
-    @property
-    def comet(self) -> comet.CometExperiment:
-        return comet.get_running_experiment()  # pyright: ignore[reportReturnType]
-
-    @functools.cached_property
-    def exp_dir(self) -> Path:
-        return _path.exp_dir(absolute=True)
-
-    @property
-    def exp_key(self) -> str:
-        return self.comet.get_key()
-
-    @property
-    def exp_name(self) -> str:
-        return self.comet.get_name()
-
-    @exp_name.setter
-    def exp_name(self, value: str) -> None:
-        self.comet.set_name(value)
-
-    @property
-    def exp_url(self) -> str:
-        return self.comet.url  # pyright: ignore[reportReturnType]
-
-    @property
-    def project_id(self) -> str:
-        return self.comet.project_id  # pyright: ignore[reportReturnType]
-
-    @property
-    def project_name(self) -> str:
-        return self.comet.project_name
-
-    @functools.cached_property
-    def start_time(self) -> datetime.datetime:
-        return datetime.datetime.now().astimezone()
-
-    def log_input(self, path: PathLike, /, **kwargs) -> None:
-        self.log_asset(path, prefix="inputs/", **kwargs)
-
-    def log_output(self, path: PathLike, /, **kwargs) -> None:
-        self.log_asset(path, prefix="outputs/", **kwargs)
-
-
-exp = Experiment()
-add_tag: AddTag = exp.add_tag
-add_tags: AddTags = exp.add_tags
-end: End = exp.end
-log_asset: LogAsset = exp.log_asset
-log_code: LogCode = exp.log_code
-log_metric: LogMetric = exp.log_metric
-log_metrics: LogMetrics = exp.log_metrics
-log_other: LogOther = exp.log_other
-log_others: LogOthers = exp.log_others
-log_param: LogParam = exp.log_param
-log_params: LogParams = exp.log_params
-start: Start = exp.start
-
-
-def current_exp() -> Experiment:
-    return exp
-
-
-def log_input(path: PathLike, /, **kwargs) -> None:
-    exp.log_input(path, **kwargs)
-
-
-def log_output(path: PathLike, /, **kwargs) -> None:
-    exp.log_output(path, **kwargs)

liblaf/cherries/integration/comet.py

@@ -1,142 +0,0 @@
-from collections.abc import Iterable, Mapping
-from pathlib import Path
-from typing import Any, override
-from unittest import mock
-
-import attrs
-import comet_ml as comet
-from loguru import logger
-
-from liblaf.cherries import meta
-from liblaf.cherries.typed import PathLike
-
-from . import _abc
-
-
-class Comet:
-    @property
-    def exp(self) -> comet.CometExperiment:
-        exp: comet.CometExperiment | None = comet.get_running_experiment()
-        if exp is None:
-            exp = mock.MagicMock()
-        return exp
-
-
-@attrs.define
-class AddTag(Comet, _abc.AddTag):
-    @override
-    def __call__(self, tag: str, **kwargs) -> None:
-        self.exp.add_tag(tag)
-
-
-@attrs.define
-class AddTags(Comet, _abc.AddTags):
-    @override
-    def __call__(self, tags: Iterable[str], **kwargs) -> None:
-        self.exp.add_tags(list(tags))
-
-
-@attrs.define
-class End(Comet, _abc.End):
-    @override
-    def __call__(self, **kwargs) -> None:
-        comet.end()
-
-
-@attrs.define
-class LogAsset(Comet, _abc.LogAsset):
-    @override
-    def __call__(self, path: PathLike, prefix: str | None = None, **kwargs) -> None:
-        path = Path(path)
-        if (dvc_path := path.with_name(path.name + ".dvc")).exists():
-            path = dvc_path
-        if path.is_file():
-            filename: str | None = f"{prefix}/{path.name}" if prefix else None
-            self.exp.log_asset(path, file_name=filename, **kwargs)
-        elif path.is_dir():
-            self.exp.log_asset_folder(str(path), **kwargs)
-        else:
-            logger.warning("Neither file nor directory: {}", path)
-
-
-@attrs.define
-class LogCode(Comet, _abc.LogCode):
-    @override
-    def __call__(self, path: PathLike, **kwargs) -> None:
-        path = Path(path)
-        if path.is_file():
-            self.exp.log_code(str(path), **kwargs)
-        elif path.is_dir():
-            self.exp.log_code(folder=str(path), **kwargs)
-        else:
-            logger.warning("Neither file nor directory: {}", path)
-
-
-@attrs.define
-class LogMetric(Comet, _abc.LogMetric):
-    @override
-    def __call__(
-        self,
-        name: str,
-        value: float,
-        step: int | None = None,
-        epoch: int | None = None,
-        **kwargs,
-    ) -> None:
-        self.exp.log_metric(name, value, step=step, epoch=epoch)
-
-
-@attrs.define
-class LogMetrics(Comet, _abc.LogMetrics):
-    @override
-    def __call__(
-        self,
-        metrics: Mapping[str, Any],
-        step: int | None = None,
-        epoch: int | None = None,
-        **kwargs,
-    ) -> None:
-        self.exp.log_metrics(metrics, step=step, epoch=epoch)  # pyright: ignore[reportArgumentType]
-
-
-@attrs.define
-class LogOther(Comet, _abc.LogOther):
-    @override
-    def __call__(self, key: str, value: Any, **kwargs) -> None:
-        self.exp.log_other(key, value)
-
-
-@attrs.define
-class LogOthers(Comet, _abc.LogOthers):
-    @override
-    def __call__(self, others: Mapping[str, Any], **kwargs) -> None:
-        self.exp.log_others(others)  # pyright: ignore[reportArgumentType]
-
-
-@attrs.define
-class LogParam(Comet, _abc.LogParam):
-    @override
-    def __call__(self, name: str, value: Any, **kwargs) -> None:
-        self.exp.log_parameter(name, value)
-
-
-@attrs.define
-class LogParams(Comet, _abc.LogParams):
-    @override
-    def __call__(self, params: Any, **kwargs: Any) -> None:
-        self.exp.log_parameters(params)
-
-
-@attrs.define
-class Start(Comet, _abc.Start):
-    text: bool = True
-    jsonl: bool = False
-
-    @override
-    def __call__(self, **kwargs) -> None:
-        comet.start(
-            project_name=meta.project_name(),
-            experiment_config=comet.ExperimentConfig(
-                auto_output_logging="native", name=meta.exp_name()
-            ),
-        )

liblaf/cherries/integration/dvc.py

@@ -1,51 +0,0 @@
-import subprocess as sp
-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
-
-from . import _abc
-
-
-@attrs.define
-class End(_abc.End):
-    priority: int = attrs.field(default=-2, kw_only=True)  # after logging
-
-    @override
-    def __call__(self, **kwargs) -> None:
-        sp.run(["dvc", "status"], check=True)
-        sp.run(["dvc", "push"], check=True)
-
-
-@attrs.define
-class LogAsset(_abc.LogAsset):
-    priority: int = attrs.field(default=-1, kw_only=True)  # before Comet
-
-    @override
-    def __call__(self, path: PathLike, prefix: str | None = None, **kwargs) -> None:
-        path: Path = _path.as_path(path)
-        if check_ignore(path) or tracked_by_git(path):
-            return
-        sp.run(["dvc", "add", "--quiet", path], check=True)
-
-
-def check_ignore(path: PathLike) -> bool:
-    proc: sp.CompletedProcess[bytes] = sp.run(
-        ["dvc", "check-ignore", "--quiet", path], check=False
-    )
-    return proc.returncode == 0
-
-
-def tracked_by_git(path: PathLike) -> bool:
-    path: Path = _path.as_path(path).absolute()
-    try:
-        repo = git.Repo(search_parent_directories=True)
-        repo.git.ls_files(path, error_unmatch=True)
-    except git.exc.GitCommandError:
-        return False
-    return True

liblaf/cherries/integration/git.py

@@ -1,44 +0,0 @@
-from typing import override
-
-import attrs
-
-from liblaf.cherries import meta as _info
-
-from . import _abc
-from ._exp import exp
-
-
-@attrs.define
-class End(_abc.End):
-    priority: int = attrs.field(default=-3, kw_only=True)  # before dvc
-    dry_run: bool = False
-
-    @override
-    def __call__(self, **kwargs) -> None:
-        git_auto_commit(
-            "chore(cherries): auto commit (on run end)", dry_run=self.dry_run
-        )
-
-
-@attrs.define
-class Start(_abc.Start):
-    dry_run: bool = False
-
-    @override
-    def __call__(self, **kwargs) -> None:
-        git_auto_commit(
-            "chore(cherries): auto commit (on run start)", dry_run=self.dry_run
-        )
-
-
-def git_auto_commit(
-    header: str = "chore(cherries): auto commit", *, dry_run: bool = False
-) -> None:
-    body: str = ""
-    if exp.exp_name and exp.exp_url:
-        body += f"🧪 View experiment {exp.exp_name} at: {exp.exp_url}\n"
-    message: str = f"{header}\n\n{body}" if body else header
-    _info.git_auto_commit(message, dry_run=dry_run)
-    exp.log_other("cherries.git.branch", _info.git_branch())
-    exp.log_other("cherries.git.sha", _info.git_commit_sha())
-    exp.log_other("cherries.git.url", _info.git_commit_url())

liblaf/cherries/integration/logging.py

@@ -1,45 +0,0 @@
-from typing import override
-
-import attrs
-import loguru
-from loguru import logger
-
-from liblaf import grapes
-
-from . import _abc
-from ._exp import exp
-
-
-@attrs.define
-class End(_abc.End):
-    priority: int = attrs.field(default=-1, kw_only=True)  # before Comet
-    text: bool = True
-    jsonl: bool = False
-
-    @override
-    def __call__(self, **kwargs) -> None:
-        logger.complete()
-        if self.text and (exp.exp_dir / "run.log").exists():
-            exp.log_asset(exp.exp_dir / "run.log")
-        if self.jsonl and (exp.exp_dir / "run.log.jsonl").exists():
-            exp.log_asset(exp.exp_dir / "run.log.jsonl")
-
-
-@attrs.define
-class Start(_abc.Start):
-    priority: int = attrs.field(default=-1, kw_only=True)  # before Comet
-    text: bool = True
-    jsonl: bool = False
-
-    @override
-    def __call__(self, **kwargs) -> None:
-        handlers: list[loguru.HandlerConfig] = [
-            grapes.logging.rich_handler(enable_link=False)
-        ]
-        if self.text:
-            handlers.append(grapes.logging.file_handler(sink=exp.exp_dir / "run.log"))
-        if self.jsonl:
-            handlers.append(
-                grapes.logging.jsonl_handler(sink=exp.exp_dir / "run.log.jsonl")
-            )
-        grapes.init_logging(handlers=handlers)

liblaf/cherries/presets/__init__.pyi

@@ -1,5 +0,0 @@
-from ._default import default
-from ._playground import playground
-from .typed import Preset
-
-__all__ = ["Preset", "default", "playground"]

liblaf/cherries/presets/_default.py

@@ -1,46 +0,0 @@
-from liblaf.cherries import integration
-
-
-def default(exp: integration.Experiment) -> integration.Experiment:
-    exp.add_tag.add(
-        integration.comet.AddTag(),
-    )
-    exp.add_tags.add(
-        integration.comet.AddTags(),
-    )
-    exp.end.add(
-        integration.git.End(),
-        integration.dvc.End(),
-        integration.logging.End(),
-        integration.comet.End(),
-    )
-    exp.log_asset.add(
-        integration.dvc.LogAsset(),
-        integration.comet.LogAsset(),
-    )
-    exp.log_code.add(
-        integration.comet.LogCode(),
-    )
-    exp.log_metric.add(
-        integration.comet.LogMetric(),
-    )
-    exp.log_metrics.add(
-        integration.comet.LogMetrics(),
-    )
-    exp.log_other.add(
-        integration.comet.LogOther(),
-    )
-    exp.log_others.add(
-        integration.comet.LogOthers(),
-    )
-    exp.log_param.add(
-        integration.comet.LogParam(),
-    )
-    exp.log_params.add(
-        integration.comet.LogParams(),
-    )
-    exp.start.add(
-        integration.logging.Start(),
-        integration.comet.Start(),
-    )
-    return integration.exp

liblaf/cherries/presets/_playground.py

@@ -1,11 +0,0 @@
-from liblaf.cherries import integration
-
-
-def playground(exp: integration.Experiment) -> integration.Experiment:
-    exp.end.add(
-        integration.logging.End(),
-    )
-    exp.start.add(
-        integration.logging.Start(),
-    )
-    return integration.exp

liblaf/cherries/presets/typed.py

@@ -1,5 +0,0 @@
-from collections.abc import Callable
-
-from liblaf.cherries import integration
-
-type Preset = Callable[[integration.Experiment], integration.Experiment]

liblaf_cherries-0.1.6.dist-info/RECORD

@@ -1,44 +0,0 @@
-liblaf/cherries/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/__init__.pyi,sha256=XAiCE1ZKUkEHQ44ElWLnmxZXxyrWU4hVH1NA4SjISUo,2022
-liblaf/cherries/_run.py,sha256=9zbivlGOUc-S4Luu14FQeRP2fNSBDca8EkSYWaES5EQ,1479
-liblaf/cherries/_version.py,sha256=ESbJO0YD7TYfOUv_WDIJJgWELGepEWsoyhqVifEcXPA,511
-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/config/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/config/__init__.pyi,sha256=1CMEQlPneeRXCG51KDUj6Zmqs0xzUF3gilqeuHPlsVc,359
-liblaf/cherries/config/_asset.py,sha256=7WEkyDxX4X389s0UqoBRbOtHOU9Dlh4NtAq9KXHXbPI,2485
-liblaf/cherries/config/_config.py,sha256=WPwwk-3O96FyHGb2W8__LDfHpXBHLQM44aOcrMPjDL4,171
-liblaf/cherries/core/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/core/__init__.pyi,sha256=E2YMg93kfxxO-g0CGyDOqkzTmVlSlHC7DRmu3AEm4_w,220
-liblaf/cherries/core/_exp.py,sha256=3kexCigujuHnO0NWSfa46FBwe0QGpAJtMW-WIphmAMA,2726
-liblaf/cherries/core/_spec.py,sha256=5rOhMgFTVhXhvOqJ4USWrYuuhOxIgyPdI5LFknpnMTs,5357
-liblaf/cherries/integration/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/integration/__init__.pyi,sha256=LSvxbLYdyYAsc8aR2OMYiEf0hku1L82q0ECqM6hk_UQ,1042
-liblaf/cherries/integration/_abc.py,sha256=0xsoAxxRAhBAOEx0qNx7THDVF8zdYuaAUHX7INe0r8U,3817
-liblaf/cherries/integration/_exp.py,sha256=gsCCqi551JP5EUHagpW29jHH1PSE7prfpMAuqKJn-9A,3048
-liblaf/cherries/integration/comet.py,sha256=BbY88hwo3rVphirYVSBSfQgs8z0UB9RNlCqK5wMUCRY,3709
-liblaf/cherries/integration/dvc.py,sha256=ClIyT-xj7jx_8knwdFKMzXH6Gq7lQnNWBzW8zycZB8g,1358
-liblaf/cherries/integration/git.py,sha256=GPRPpftlkDfb1z4oSQ6gRhnbPowKB0euOK8Ak4oAJpw,1209
-liblaf/cherries/integration/logging.py,sha256=lIjF_2PE6TGT-uFXAnQOnPZ4kP-e982yrGL03L4Gn3s,1267
-liblaf/cherries/meta/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/meta/__init__.pyi,sha256=kQFneP3IiV9rBIzpep_uX0z-5IRPrXkPmyNRt19j8fg,282
-liblaf/cherries/meta/_git.py,sha256=oy1xx23sXWA7RqQuLGc3HxcLCF7zHTSpZvZ9pukJGdI,1232
-liblaf/cherries/meta/_name.py,sha256=LskfsW0yv9WEAl8WqGFG3D82_mULulV5N9p9Ogjt7Q0,561
-liblaf/cherries/pathutils/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/pathutils/__init__.pyi,sha256=mr3Dk8ragpNGXmXJriMt21vc7YmqUD6X587exFe9tCI,413
-liblaf/cherries/pathutils/_convert.py,sha256=IrXN_9s-rfH7EKk0_ilr1giGmfN8FlvB_edm2kI-k3I,807
-liblaf/cherries/pathutils/_path.py,sha256=PLcVzBO6r99qspf5ZKqVtWm6mT8Ud5cUGsUkyJkTqyM,1548
-liblaf/cherries/pathutils/_special.py,sha256=b5D3PtW__u0Zpiv69OYoV7TTYc6Dgs4Bu-P8jwrROV4,1415
-liblaf/cherries/presets/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/presets/__init__.pyi,sha256=3ug6RF4DSBOEankXOEgotQw_E0fMvQ5yuhps7EHf6b0,139
-liblaf/cherries/presets/_default.py,sha256=-sm9IBvXP_JfWXUfZdwc9u5WcLOQXloDmBgwjiw2EPk,1095
-liblaf/cherries/presets/_playground.py,sha256=F0dphGgJ_6MD5mBGm0chrxWIufZtYgtt5c2l-TVPNso,260
-liblaf/cherries/presets/typed.py,sha256=LztTXKiiCD1NW1fb2Od6giaU0vaXdOidjsqxbTkIlJY,152
-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.1.6.dist-info/METADATA,sha256=CFvkztVl4GVLIuIgCvciRnt32SmNZcudEhRRPmGit9U,6194
-liblaf_cherries-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-liblaf_cherries-0.1.6.dist-info/licenses/LICENSE,sha256=Ph4NzyU3lGVDeYv-mf8aRmImH8v9rVL9F362FV4G6Ow,1063
-liblaf_cherries-0.1.6.dist-info/RECORD,,