liblaf-cherries 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

liblaf/cherries/__init__.pyi

@@ -1,7 +1,24 @@
-from . import config, core, meta, paths
+from . import config, core, meta, paths, plugins
 from ._entrypoint import end, run, start
 from .config import BaseConfig, input, output  # noqa: A004
-from .core import Plugin, Run, active_run, log_asset, log_metrics
+from .core import (
+    Plugin,
+    Run,
+    active_run,
+    log_asset,
+    log_asset_data,
+    log_asset_folder,
+    log_input,
+    log_input_folder,
+    log_metric,
+    log_metrics,
+    log_other,
+    log_others,
+    log_output,
+    log_output_folder,
+    log_parameter,
+    log_parameters,
+)
 from .paths import (
     as_os_path,
     as_path,
@@ -37,13 +54,25 @@ __all__ = [
     "input",
     "inputs",
     "log_asset",
+    "log_asset_data",
+    "log_asset_folder",
+    "log_input",
+    "log_input_folder",
+    "log_metric",
     "log_metrics",
+    "log_other",
+    "log_others",
+    "log_output",
+    "log_output_folder",
+    "log_parameter",
+    "log_parameters",
     "meta",
     "output",
     "outputs",
     "params",
     "path",
     "paths",
+    "plugins",
     "run",
     "src",
     "start",

liblaf/cherries/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.1'
-__version_tuple__ = version_tuple = (0, 2, 1)
+__version__ = version = '0.2.2'
+__version_tuple__ = version_tuple = (0, 2, 2)

liblaf/cherries/core/__init__.pyi

@@ -1,6 +1,24 @@
 from ._impl import ImplInfo, get_impl_info, impl
 from ._plugin import Plugin
-from ._run import Run, active_run, log_asset, log_metrics
+from ._run import (
+    Run,
+    active_run,
+    end,
+    log_asset,
+    log_asset_data,
+    log_asset_folder,
+    log_input,
+    log_input_folder,
+    log_metric,
+    log_metrics,
+    log_other,
+    log_others,
+    log_output,
+    log_output_folder,
+    log_parameter,
+    log_parameters,
+    start,
+)
 from ._spec import SpecInfo, spec
 from .typed import MethodName, PluginId
 
@@ -12,9 +30,22 @@ __all__ = [
     "Run",
     "SpecInfo",
     "active_run",
+    "end",
     "get_impl_info",
     "impl",
     "log_asset",
+    "log_asset_data",
+    "log_asset_folder",
+    "log_input",
+    "log_input_folder",
+    "log_metric",
     "log_metrics",
+    "log_other",
+    "log_others",
+    "log_output",
+    "log_output_folder",
+    "log_parameter",
+    "log_parameters",
     "spec",
+    "start",
 ]

liblaf/cherries/core/_impl.py

@@ -3,6 +3,7 @@ from collections.abc import Callable, Iterable
 from typing import Any, overload
 
 import attrs
+import wrapt
 
 from liblaf import grapes
 
@@ -41,18 +42,19 @@ def impl(
 ) -> Any:
     if func is None:
         return functools.partial(impl, priority=priority, after=after, before=before)
-    info = ImplInfo(after=after, before=before, priority=priority)
 
-    @grapes.decorator(attrs={"_self_impl": info})
+    @wrapt.decorator
     def wrapper(
         wrapped: Callable, _instance: Any, args: tuple, kwargs: dict[str, Any]
     ) -> Any:
         return wrapped(*args, **kwargs)
 
-    return wrapper(func)
+    proxy: Any = wrapper(func)  # pyright: ignore[reportCallIssue]
+    proxy._self_impl = ImplInfo(after=after, before=before, priority=priority)  # noqa: SLF001
+    return proxy
 
 
 def get_impl_info(func: Callable | None) -> ImplInfo | None:
     if func is None:
         return None
-    return getattr(func, "_self_impl", None)
+    return grapes.unbind_getattr(func, "_self_impl", None)

liblaf/cherries/core/_plugin.py

@@ -1,4 +1,3 @@
-import inspect
 from collections.abc import Mapping, MutableMapping, Sequence
 from typing import Any, Self
 
@@ -6,7 +5,7 @@ import attrs
 import networkx as nx
 
 from ._impl import ImplInfo, get_impl_info
-from ._spec import SpecInfo
+from ._spec import SpecInfo, collect_specs
 from .typed import MethodName
 
 
@@ -33,15 +32,6 @@ class Plugin:
             return self
         return self._plugin_parent.plugin_root
 
-    @property
-    def specs(self) -> dict[str, SpecInfo]:
-        return {
-            name: method._self_spec  # noqa: SLF001
-            for name, method in inspect.getmembers(
-                type(self), lambda m: getattr(m, "_self_spec", None) is not None
-            )
-        }
-
     def delegate(
         self,
         method: MethodName,
@@ -65,7 +55,8 @@ class Plugin:
         self.plugins[plugin.plugin_id] = plugin
 
     def _prepare(self) -> None:
-        for method in self.specs:
+        specs: dict[str, SpecInfo] = collect_specs(self)
+        for method in specs:
             self._sort_plugins_cache[method] = self._sort_plugins(
                 method, refresh_cache=True
             )

liblaf/cherries/core/_run.py

@@ -1,4 +1,6 @@
+from collections.abc import Mapping
 from pathlib import Path
+from typing import Any
 
 import attrs
 
@@ -22,14 +24,65 @@ class Run(Plugin):
     def exp_dir(self) -> Path:
         return paths.exp_dir(absolute=True)
 
+    @property
+    def url(self) -> str:
+        return self.get_url()
+
     @spec
     def end(self, *args, **kwargs) -> None: ...
-
+    @spec(first_result=True)
+    def get_url(self) -> str: ...
     @spec
     def log_asset(self, *args, **kwargs) -> None: ...
-
     @spec
-    def log_metrics(self, *args, **kwargs) -> None: ...
+    def log_asset_data(self, *args, **kwargs) -> None: ...
+    @spec
+    def log_asset_folder(self, *args, **kwargs) -> None: ...
+    @spec
+    def log_input(self, *args, **kwargs) -> None: ...
+    @spec
+    def log_input_folder(self, *args, **kwargs) -> None: ...
+    @spec
+    def log_metric(
+        self,
+        name: str,
+        value: Any,
+        /,
+        step: int | None = None,
+        epoch: int | None = None,
+        **kwargs,
+    ) -> None: ...
+    @spec
+    def log_metrics(
+        self,
+        dic: Mapping[str, Any],
+        /,
+        prefix: str | None = None,
+        step: int | None = None,
+        epoch: int | None = None,
+        **kwargs,
+    ) -> None: ...
+    @spec
+    def log_other(self, key: Any, value: Any, /, **kwargs) -> None: ...
+    @spec
+    def log_others(self, dictionary: Mapping[Any, Any], /, **kwargs) -> None: ...
+    @spec
+    def log_output(self, *args, **kwargs) -> None: ...
+    @spec
+    def log_output_folder(self, *args, **kwargs) -> None: ...
+    @spec
+    def log_parameter(
+        self, name: Any, value: Any, /, step: int | None = None, **kwargs
+    ) -> None: ...
+    @spec
+    def log_parameters(
+        self,
+        parameters: Mapping[Any, Any],
+        /,
+        prefix: str | None = None,
+        step: int | None = None,
+        **kwargs,
+    ) -> None: ...
 
     @spec(delegate=False)
     def start(self, *args, **kwargs) -> None:
@@ -38,5 +91,18 @@ class Run(Plugin):
 
 
 active_run: Run = Run()
+end = active_run.end
 log_asset = active_run.log_asset
+log_asset_data = active_run.log_asset_data
+log_asset_folder = active_run.log_asset_folder
+log_input = active_run.log_input
+log_input_folder = active_run.log_input_folder
+log_metric = active_run.log_metric
 log_metrics = active_run.log_metrics
+log_other = active_run.log_other
+log_others = active_run.log_others
+log_output = active_run.log_output
+log_output_folder = active_run.log_output_folder
+log_parameter = active_run.log_parameter
+log_parameters = active_run.log_parameters
+start = active_run.start

liblaf/cherries/core/_spec.py

@@ -4,6 +4,7 @@ from collections.abc import Callable, Mapping, Sequence
 from typing import Any, Protocol, overload
 
 import attrs
+import wrapt
 
 from liblaf import grapes
 
@@ -47,7 +48,7 @@ def spec(
 
     info = SpecInfo(delegate=delegate, first_result=first_result)
 
-    @grapes.decorator(attrs={"_self_spec": info})
+    @wrapt.decorator
     def wrapper(
         wrapped: Callable, instance: Plugin, args: tuple, kwargs: dict[str, Any]
     ) -> Any:
@@ -57,15 +58,17 @@ def spec(
             )
         return wrapped(*args, **kwargs)
 
-    return wrapper(func)
+    proxy: Any = wrapper(func)  # pyright: ignore[reportCallIssue]
+    proxy._self_spec = SpecInfo(delegate=delegate, first_result=first_result)  # noqa: SLF001
+    return proxy
 
 
 def collect_specs(cls: type[Plugin] | Plugin) -> dict[str, SpecInfo]:
     if isinstance(cls, type):
         cls = type(cls)
     return {
-        name: method._self_spec  # noqa: SLF001
+        name: grapes.unbind_getattr(method, "_self_spec")
         for name, method in inspect.getmembers(
-            cls, lambda m: getattr(m, "_self_spec", None) is not None
+            cls, lambda m: grapes.unbind_getattr(m, "_self_spec", None) is not None
         )
     }

liblaf/cherries/plugins/__init__.pyi

@@ -1,3 +1,4 @@
+from .comet import Comet
 from .logging import Logging
 
-__all__ = ["Logging"]
+__all__ = ["Comet", "Logging"]

liblaf/cherries/plugins/comet.py

@@ -0,0 +1,54 @@
+from typing import override
+
+import comet_ml
+
+from liblaf.cherries import core
+
+
+class Comet(core.Run):
+    exp: comet_ml.CometExperiment
+
+    @override
+    @core.impl(after=("Logging",))
+    def end(self, *args, **kwargs) -> None:
+        return self.exp.end()
+
+    @override
+    @core.impl
+    def get_url(self) -> str:
+        return self.exp.url  # pyright: ignore[reportReturnType]
+
+    @override
+    @core.impl
+    def log_metric(self, *args, **kwargs) -> None:
+        return self.exp.log_metric(*args, **kwargs)
+
+    @override
+    @core.impl
+    def log_metrics(self, *args, **kwargs) -> None:
+        return self.exp.log_metrics(*args, **kwargs)
+
+    @override
+    @core.impl
+    def log_other(self, *args, **kwargs) -> None:
+        return self.exp.log_other(*args, **kwargs)
+
+    @override
+    @core.impl
+    def log_others(self, *args, **kwargs) -> None:
+        return self.exp.log_others(*args, **kwargs)
+
+    @override
+    @core.impl
+    def log_parameter(self, *args, **kwargs) -> None:
+        return self.exp.log_parameter(*args, **kwargs)
+
+    @override
+    @core.impl
+    def log_parameters(self, *args, **kwargs) -> None:
+        return self.exp.log_parameters(*args, **kwargs)
+
+    @override
+    @core.impl(after=("Logging",))
+    def start(self, *args, **kwargs) -> None:
+        self.exp = comet_ml.start()

liblaf/cherries/plugins/logging.py

@@ -10,7 +10,8 @@ class Logging(core.Run):
     def start(self, *args, **kwargs) -> None:
         profile = grapes.logging.profiles.ProfileCherries(
             handlers=[
-                grapes.logging.rich_handler(),
+                # Comet and many other services do not support links
+                grapes.logging.rich_handler(enable_link=False),
                 grapes.logging.file_handler(sink=self.plugin_root.exp_dir / "run.log"),
             ]
         )

liblaf/cherries/profiles/_default.py

@@ -1,6 +1,6 @@
 from typing import override
 
-from liblaf.cherries import core
+from liblaf.cherries import core, plugins
 
 from ._playground import ProfilePlayground
 
@@ -9,4 +9,5 @@ class ProfileDefault(ProfilePlayground):
     @override  # impl Profile
     def init(self) -> core.Run:
         run: core.Run = super().init()
+        run.register(plugins.Comet())
         return run

{liblaf_cherries-0.2.1.dist-info → liblaf_cherries-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: liblaf-cherries
-Version: 0.2.1
+Version: 0.2.2
 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/
@@ -37,7 +37,7 @@ Requires-Dist: dvc[webdav]<4,>=3.0.0
 Requires-Dist: environs<15,>=14.0.0
 Requires-Dist: gitpython<4,>=3.0.0
 Requires-Dist: lazy-loader<0.5,>=0.4
-Requires-Dist: liblaf-grapes<1.1,>=1
+Requires-Dist: liblaf-grapes<2,>=1.0.0
 Requires-Dist: loguru<0.8,>=0.7.0
 Requires-Dist: networkx<4,>=3.0.0
 Requires-Dist: pydantic-settings<3,>=2.0.0

{liblaf_cherries-0.2.1.dist-info → liblaf_cherries-0.2.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 liblaf/cherries/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/__init__.pyi,sha256=urWKt4Q2hPNdZK-eLO1fkqqbIJ0dtiHTb6QYWgDojkA,823
+liblaf/cherries/__init__.pyi,sha256=_4VWvFPs2Kpr-u1tO_YU8SmFdhZPNUmOvtJQo9qXn4Q,1302
 liblaf/cherries/_entrypoint.py,sha256=bBjzM770-9lybNq0qtSYdiIPsWIVoHttgkxj02myyAU,1581
-liblaf/cherries/_version.py,sha256=UoNvMtd4wCG76RwoSpNCUtaFyTwakGcZolfjXzNVSMY,511
+liblaf/cherries/_version.py,sha256=OjGGK5TcHVG44Y62aAqeJH4CskkZoY9ydbHOtCDew50,511
 liblaf/cherries/_version.pyi,sha256=Pnv4Bxw13LHeuVkPLPsTtnp4N4jOGcAfFJw05uMMgBY,108
 liblaf/cherries/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 liblaf/cherries/typed.py,sha256=mim8QVtwczTSHyw5mhEdfFcXis9o32n0CZyu8BrEorE,50
@@ -10,11 +10,11 @@ liblaf/cherries/config/__init__.pyi,sha256=1CMEQlPneeRXCG51KDUj6Zmqs0xzUF3gilqeu
 liblaf/cherries/config/_asset.py,sha256=OZQ521GIOWNWMdcF_iV3t5ZG5XhdlaQ6_diuqP_DPx0,2472
 liblaf/cherries/config/_config.py,sha256=WPwwk-3O96FyHGb2W8__LDfHpXBHLQM44aOcrMPjDL4,171
 liblaf/cherries/core/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/core/__init__.pyi,sha256=3hAxoGwuAYfJs5UzYGiyZSwA4nYNarGziceK7jh-CLs,414
-liblaf/cherries/core/_impl.py,sha256=1sRAIdWNB5eLJ45cZOdLn0VZ06mqkxI6r5GXSMXoUWo,1374
-liblaf/cherries/core/_plugin.py,sha256=mZxki48dC64CxsOj5dZiKlSPZG9zioun_pP0y3AHtZo,3138
-liblaf/cherries/core/_run.py,sha256=O8B2aOZwxdnwBfAv24nA7FkFn7KgLF9EjFkeQs6hcrQ,1020
-liblaf/cherries/core/_spec.py,sha256=b6sTw5C08dnAjw_YRcGyd32XbSTvGMekkzE_qPvZI9g,1789
+liblaf/cherries/core/__init__.pyi,sha256=9jbA822Znhm9AHXR_zhcM74e8pikNAdkkSpNUOjKQjI,909
+liblaf/cherries/core/_impl.py,sha256=LCd4f5oX5bEMa4uX37r0yiPntqZM1Lwqrd7vKL1oc7w,1459
+liblaf/cherries/core/_plugin.py,sha256=dBK5re9yAB8kfp4Uc5fu5aVLms_lMi2bB4jDJEayvfs,2905
+liblaf/cherries/core/_run.py,sha256=3VMT4fVIoWU5FYss5SzKgWAr9vcYWwbrQCiMh7aeG44,2972
+liblaf/cherries/core/_spec.py,sha256=F73wxygkc_3v0hOKwfFSGaLRHEDhhENWLQZakruuRq8,1950
 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
@@ -26,18 +26,19 @@ liblaf/cherries/paths/_convert.py,sha256=IrXN_9s-rfH7EKk0_ilr1giGmfN8FlvB_edm2kI
 liblaf/cherries/paths/_path.py,sha256=Wz3C4ZgERzP659I_eC1N9lMokyyvbIedCMs3PpYrYus,1606
 liblaf/cherries/paths/_special.py,sha256=b5D3PtW__u0Zpiv69OYoV7TTYc6Dgs4Bu-P8jwrROV4,1415
 liblaf/cherries/plugins/__init__.py,sha256=OHb6Xou2v6u42swTgjRfzej4CIlRg4OmgOIQXUiRjKA,97
-liblaf/cherries/plugins/__init__.pyi,sha256=PUOon4osKSfGEQCj6NG2wUgwCOh4G8LcHfOad_RFs6o,52
-liblaf/cherries/plugins/logging.py,sha256=_Dzzc-jBWIvWWhfq_i-dq2IsX1Vnan0cDNy7Sar_Uvc,836
+liblaf/cherries/plugins/__init__.pyi,sha256=80ZJM2XNFDGjclyCzmlTia0nzDPuOADV8jirU8ZGc_g,86
+liblaf/cherries/plugins/comet.py,sha256=R2rUS2n6ynSqKwADRS7ZKij09DZtUJ2Lm8LbsCB9P3U,1331
+liblaf/cherries/plugins/logging.py,sha256=pZut8_TnCZscjkyRpTcw3PWhL1Vv1FJ60f-XcpSNPbI,922
 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=7V0BkyB80iVOI7bray4rWHGYlMqN3oEzAg3eOJ2XRWI,269
+liblaf/cherries/profiles/_default.py,sha256=8fsEAN6s9ZDxngX1PMb9JISIfC6pilEWYgynuXn5TxY,316
 liblaf/cherries/profiles/_factory.py,sha256=FhYfHBmWe0uPOhRmCCg8zilUggHIPIllrPDx4Gkc8C8,602
 liblaf/cherries/profiles/_playground.py,sha256=QS_YjYu5ewloF3_U4dZFmqCHx3WVHyIKcqsVe1WbYrc,295
 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.2.1.dist-info/METADATA,sha256=YZ8wRa6oe2uwraxGLT_N-8kZQBh46V4_lBIMpeTfgjU,6252
-liblaf_cherries-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-liblaf_cherries-0.2.1.dist-info/licenses/LICENSE,sha256=Ph4NzyU3lGVDeYv-mf8aRmImH8v9rVL9F362FV4G6Ow,1063
-liblaf_cherries-0.2.1.dist-info/RECORD,,
+liblaf_cherries-0.2.2.dist-info/METADATA,sha256=KMA-Ya-Ao2z-V5kSEXUKB4_iFkIJtfJtOS83KF97ZsE,6254
+liblaf_cherries-0.2.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+liblaf_cherries-0.2.2.dist-info/licenses/LICENSE,sha256=Ph4NzyU3lGVDeYv-mf8aRmImH8v9rVL9F362FV4G6Ow,1063
+liblaf_cherries-0.2.2.dist-info/RECORD,,