liblaf-cherries 0.4.3__py3-none-any.whl → 0.5.1__py3-none-any.whl

liblaf/cherries/config/_asset.py

@@ -1,92 +0,0 @@
-import enum
-from collections.abc import Callable, Iterable
-from pathlib import Path
-from typing import Any, Literal
-
-import pydantic
-
-from liblaf import grapes
-from liblaf.cherries import paths
-from liblaf.cherries.typed import PathLike
-
-
-class AssetKind(enum.StrEnum):
-    INPUT = enum.auto()
-    OUTPUT = enum.auto()
-
-
-type PathGenerator = (
-    PathLike | Iterable[PathLike] | Callable[[PathLike], PathLike | Iterable[PathLike]]
-)
-
-
-class MetaAsset:
-    kind: AssetKind
-    _extra: PathGenerator | None = None
-
-    def __init__(self, kind: AssetKind, extra: PathGenerator | None = None) -> None:
-        self.kind = kind
-        self._extra = extra
-
-    def get_extra(self, value: Path) -> list[Path]:
-        if self._extra is None:
-            return []
-        if callable(self._extra):
-            extra: Iterable[PathLike] = grapes.as_iterable(self._extra(value))
-            return [Path(p) for p in extra]
-        extra: Iterable[PathLike] = grapes.as_iterable(self._extra)
-        return [Path(p) for p in extra]
-
-
-def get_assets(cfg: pydantic.BaseModel, kind: AssetKind) -> list[Path]:
-    assets: list[Path] = []
-    for name, info in type(cfg).model_fields.items():
-        value: Any = getattr(cfg, name)
-        if isinstance(value, pydantic.BaseModel):
-            assets.extend(get_assets(value, kind))
-        for meta in info.metadata:
-            if isinstance(meta, MetaAsset) and meta.kind == kind:
-                value: Path = Path(value)
-                assets.append(value)
-                assets.extend(meta.get_extra(value))
-    return assets
-
-
-def get_inputs(cfg: pydantic.BaseModel) -> list[Path]:
-    return get_assets(cfg, AssetKind.INPUT)
-
-
-def get_outputs(cfg: pydantic.BaseModel) -> list[Path]:
-    return get_assets(cfg, AssetKind.OUTPUT)
-
-
-def input(path: PathLike, extra: PathGenerator | None = None, **kwargs) -> Path:  # noqa: A001
-    field_info: pydantic.fields.FieldInfo = pydantic.Field(paths.data(path), **kwargs)  # pyright: ignore[reportAssignmentType]
-    field_info.metadata.append(MetaAsset(kind=AssetKind.INPUT, extra=extra))
-    return field_info  # pyright: ignore[reportReturnType]
-
-
-def model_dump_without_assets(
-    model: pydantic.BaseModel,
-    *,
-    mode: str | Literal["json", "python"] = "json",  # noqa: PYI051
-    **kwargs,
-) -> dict[str, Any]:
-    data: dict[str, Any] = model.model_dump(mode=mode, **kwargs)
-    for name, info in type(model).model_fields.items():
-        value: Any = getattr(model, name)
-        if isinstance(value, pydantic.BaseModel):
-            value = model_dump_without_assets(value)
-        for meta in info.metadata:
-            if isinstance(meta, MetaAsset):
-                del data[name]
-                break
-        else:
-            data[name] = value
-    return data
-
-
-def output(path: PathLike, extra: PathGenerator | None = None, **kwargs) -> Path:
-    field_info: pydantic.fields.FieldInfo = pydantic.Field(paths.data(path), **kwargs)  # pyright: ignore[reportAssignmentType]
-    field_info.metadata.append(MetaAsset(kind=AssetKind.OUTPUT, extra=extra))
-    return field_info  # pyright: ignore[reportReturnType]

liblaf/cherries/paths/_path.py

@@ -1,70 +0,0 @@
-import sys
-from collections.abc import Container
-from pathlib import Path
-
-import git
-import git.exc
-from loguru import logger
-
-from liblaf.cherries import utils
-
-
-@utils.cache
-def entrypoint(*, absolute: bool = True) -> Path:
-    if absolute:
-        return _entrypoint_absolute()
-    return _entrypoint_relative()
-
-
-@utils.cache
-def git_root() -> Path:
-    entrypoint: Path = _entrypoint_absolute()
-    repo = git.Repo(entrypoint, search_parent_directories=True)
-    return Path(repo.working_dir)
-
-
-@utils.cache
-def git_root_safe() -> Path:
-    try:
-        return git_root()
-    except git.exc.InvalidGitRepositoryError:
-        logger.warning("Not in a git repository, using current directory", once=True)
-        return _entrypoint_absolute().parent
-
-
-@utils.cache
-def exp_dir(*, absolute: bool = True) -> Path:
-    if absolute:
-        return _exp_dir_absolute()
-    return _exp_dir_relative()
-
-
-@utils.cache
-def _entrypoint_absolute() -> Path:
-    path = Path(sys.argv[0])
-    return path.absolute()
-
-
-@utils.cache
-def _entrypoint_relative() -> Path:
-    path: Path = _entrypoint_absolute()
-    return path.relative_to(git_root_safe())
-
-
-EXP_DIR_NAMES: Container[str] = {"exp", "experiment", "experiments", "exps", "src"}
-
-
-@utils.cache
-def _exp_dir_absolute() -> Path:
-    entrypoint: Path = _entrypoint_absolute()
-    parent: Path = entrypoint.parent
-    if parent.name in EXP_DIR_NAMES:
-        return parent.parent
-    return parent
-
-
-@utils.cache
-def _exp_dir_relative() -> Path:
-    absolute: Path = _exp_dir_absolute()
-    root: Path = git_root_safe()
-    return absolute.relative_to(root)

liblaf_cherries-0.4.3.dist-info/RECORD

@@ -1,48 +0,0 @@
-liblaf/cherries/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/__init__.pyi,sha256=vQIYxi3iwe0Uij8Q8RCFlzN5MCCDuWgcnyox01C_nJo,1066
-liblaf/cherries/_entrypoint.py,sha256=pSRbOCuMhUnYrAsZTEideVbUDEmGocvKlMgGt_MX_Z0,2262
-liblaf/cherries/_version.py,sha256=bmI9ViMEsJ1Rjce-6ExwiNh2B7sZKTyBkze4k8NsTrU,704
-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=Tl-GCtMEtQOcYC5poH3qzQxR91A1RcjYDP3fEdJ9Qzc,425
-liblaf/cherries/config/_asset.py,sha256=pKHlzjOBc5yso3zJ18QfPzMwNETq1557E13qberoWZ0,3119
-liblaf/cherries/config/_config.py,sha256=WPwwk-3O96FyHGb2W8__LDfHpXBHLQM44aOcrMPjDL4,171
-liblaf/cherries/core/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/core/__init__.pyi,sha256=jgmGgzlTDlyFKHnSjNw3g22ZBuBQ07NmeRiq4qKZFdg,727
-liblaf/cherries/core/_impl.py,sha256=LCd4f5oX5bEMa4uX37r0yiPntqZM1Lwqrd7vKL1oc7w,1459
-liblaf/cherries/core/_plugin.py,sha256=LcCxNQkkInOOrPnSd5ZwKhnKQiUl4T-YicE9PEvF3XI,3198
-liblaf/cherries/core/_run.py,sha256=GNJEHGldjbvURkN5nAy8K9u1mWwxYHd-vkbnvfaOw8U,5403
-liblaf/cherries/core/_spec.py,sha256=F73wxygkc_3v0hOKwfFSGaLRHEDhhENWLQZakruuRq8,1950
-liblaf/cherries/core/_utils.py,sha256=WpS79yqTH0Kl9ewzCIAiUcxamHKzJOB7ZWhjq_13bzY,572
-liblaf/cherries/core/typed.py,sha256=razpiUtLAGFD9J4H5RbIEHKEXWzxFHFjtOBBRllhea4,42
-liblaf/cherries/meta/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/meta/__init__.pyi,sha256=kQFneP3IiV9rBIzpep_uX0z-5IRPrXkPmyNRt19j8fg,282
-liblaf/cherries/meta/_git.py,sha256=HhxqwKY52KahvAYU0R3BTRNT7lMJIjJLpnuSYfm02W4,1219
-liblaf/cherries/meta/_name.py,sha256=7sUZMYpGzW8ZKSNPITybrjnD2vLY57O91aK6Gd7Ud5M,550
-liblaf/cherries/paths/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/paths/__init__.pyi,sha256=Xf6-Igmj2K8hmLsJ3sXCj2Lrm0gIWcbARQB4oMP1zps,367
-liblaf/cherries/paths/_convert.py,sha256=IrXN_9s-rfH7EKk0_ilr1giGmfN8FlvB_edm2kI-k3I,807
-liblaf/cherries/paths/_path.py,sha256=36T4eE8OdT83LDwEaKjcBWS5r4jZEGBXFxeJqiB7-IQ,1604
-liblaf/cherries/paths/_special.py,sha256=HVmH6lCnj1TVzjAEmO93MGMTQi7JQWss4sHSNMieczY,1100
-liblaf/cherries/plugins/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/plugins/__init__.pyi,sha256=dyTB5ZS78Kg_7oWeChk_h7Ry_gU9k1sDiL5YOmnoG7I,177
-liblaf/cherries/plugins/comet.py,sha256=DzQ-ktyFO6JeV1di89J3P38Ok4X07mTVtAgUZDm0FLc,3607
-liblaf/cherries/plugins/dvc.py,sha256=2HslDy_8gcFehly2CNCSZpRs0Wet7RTGMwbxxlo9IVc,1052
-liblaf/cherries/plugins/git_.py,sha256=pcJDYUl3D11h4_v0paEH2SyfswTmicAjluWoWJpeJjM,2194
-liblaf/cherries/plugins/local.py,sha256=WHBJo6HIJOoTBRedd2a6nKoTJsaSUxhGN0jSAV94de8,1714
-liblaf/cherries/plugins/logging.py,sha256=A2-fd7H96Jtg1eOOmMYp7AmT3133vDHd4HAklhjH6n0,533
-liblaf/cherries/profiles/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/profiles/__init__.pyi,sha256=qXxy2LOG9hE0LKCnECdJSv2VoHhOTMVDE3sUKIuZKmw,292
-liblaf/cherries/profiles/_abc.py,sha256=1tpRrocBZNHonWaj3a264GnL5UoGS_HqU06aNZpruqY,193
-liblaf/cherries/profiles/_default.py,sha256=cfV003HBA5aGAZkgDDaHAyongFcxCWYsEFpSQNICcMs,440
-liblaf/cherries/profiles/_factory.py,sha256=d-PaE8JYllkzcqQDiyi5aP52lFJOkFdLai3XXgSKAE8,758
-liblaf/cherries/profiles/_playground.py,sha256=Aru-7RVoxNhomPLUxDLiM5wD5ZCPLy5Eymw0xQL2F9g,384
-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.4.3.dist-info/METADATA,sha256=l4dJ7883MyxpE1TkIigzEYAIpjWKWRKiJ2UJx6tVC3g,7023
-liblaf_cherries-0.4.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-liblaf_cherries-0.4.3.dist-info/licenses/LICENSE,sha256=Ph4NzyU3lGVDeYv-mf8aRmImH8v9rVL9F362FV4G6Ow,1063
-liblaf_cherries-0.4.3.dist-info/RECORD,,