laco-torch 1.0.0__tar.gz

laco_torch-1.0.0/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: laco-torch
+Version: 1.0.0
+Summary: PyTorch optimizer and scheduler configurables for laco.
+Author-email: Kurt Stolle <kurt@khws.io>
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+Requires-Dist: laco>=1.0.0
+Requires-Dist: torch>=2.0
+
+# Laco-Torch
+
+PyTorch optimizer and scheduler configurables for laco.
+
+Part of the [laco](https://github.com/khwstolle/laco) project — see the [root README](../../README.md) for an overview.
+
+## Installation
+
+```bash
+pip install laco-torch
+```
+
+## Features
+
+Auto-registers all `torch.optim` and `lr_scheduler` classes as `@L.configurable`; `register()`, `registered_names()`
+
+## Usage
+
+See [`docs/index.md`](docs/index.md) for the full guide.

laco_torch-1.0.0/README.md

@@ -0,0 +1,19 @@
+# Laco-Torch
+
+PyTorch optimizer and scheduler configurables for laco.
+
+Part of the [laco](https://github.com/khwstolle/laco) project — see the [root README](../../README.md) for an overview.
+
+## Installation
+
+```bash
+pip install laco-torch
+```
+
+## Features
+
+Auto-registers all `torch.optim` and `lr_scheduler` classes as `@L.configurable`; `register()`, `registered_names()`
+
+## Usage
+
+See [`docs/index.md`](docs/index.md) for the full guide.

laco_torch-1.0.0/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["setuptools>=75", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "laco-torch"
+version = "1.0.0"
+description = "PyTorch optimizer and scheduler configurables for laco."
+readme = "README.md"
+requires-python = ">=3.13"
+authors = [{ name = "Kurt Stolle", email = "kurt@khws.io" }]
+dependencies = ["laco>=1.0.0", "torch>=2.0"]
+
+[project.entry-points."laco.plugins"]
+torch = "laco.integrations.torch:_bootstrap"
+
+[tool.setuptools.packages.find]
+where = ["sources"]
+namespaces = true
+
+[tool.uv.sources]
+laco = { workspace = true }

laco_torch-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

laco_torch-1.0.0/sources/laco/integrations/torch/__init__.py

@@ -0,0 +1,33 @@
+"""PyTorch optimizer and scheduler configurables for laco.
+
+Importing this module registers all ``torch.optim`` optimizers and
+``torch.optim.lr_scheduler`` schedulers as ``@L.configurable`` so they
+can be used inside ``L.trace()`` and composed as laco config nodes.
+
+Examples
+--------
+::
+
+    import laco.language as L
+    import laco.integrations.torch  # registers configurables
+
+    # Record a config without constructing anything:
+    cfg = L.trace(lambda: torch.optim.AdamW(lr=1e-3, weight_decay=0.01))
+
+    # Instantiate later with model parameters:
+    optimizer = laco.instantiate(cfg, params=model.parameters())
+
+    # Schedulers work the same way:
+    sched_cfg = L.trace(lambda: torch.optim.lr_scheduler.CosineAnnealingLR(T_max=100))
+    scheduler = laco.instantiate(sched_cfg, optimizer=optimizer)
+
+All public optimizer and scheduler classes found in those namespaces at
+import time are registered. Third-party optimizers can be registered
+manually via :func:`register`.
+"""
+
+from laco.integrations.torch._core import _bootstrap as _bootstrap
+from laco.integrations.torch._core import register as register
+from laco.integrations.torch._core import registered_names as registered_names
+
+__all__ = ["_bootstrap", "register", "registered_names"]

laco_torch-1.0.0/sources/laco/integrations/torch/_core.py

@@ -0,0 +1,81 @@
+"""Implementation for laco-torch."""
+
+from __future__ import annotations
+
+import typing
+
+_REGISTRY: dict[str, type] = {}
+
+
+def register(cls: type) -> type:
+    """Register *cls* as an ``@L.configurable`` target.
+
+    Called automatically for all ``torch.optim`` and
+    ``torch.optim.lr_scheduler`` classes when ``laco.integrations.torch`` is
+    imported. Call it manually to register third-party optimizer classes.
+
+    Parameters
+    ----------
+    cls : type
+        A class to make traceable via ``L.trace()``.
+
+    Returns
+    -------
+    type
+        The class unchanged (usable as a decorator).
+    """
+    from laco.language import configurable
+
+    wrapped = configurable(cls)
+    _REGISTRY[f"{cls.__module__}.{cls.__qualname__}"] = wrapped
+    # Patch the class into its own module so existing references pick it up.
+    module = typing.cast(
+        typing.Any, __import__(cls.__module__, fromlist=[cls.__qualname__])
+    )
+    setattr(module, cls.__qualname__, wrapped)
+    return wrapped
+
+
+def registered_names() -> list[str]:
+    """Return the sorted fully-qualified names of all registered configurables.
+
+    Returns
+    -------
+    list[str]
+        Sorted list of ``"module.ClassName"`` strings.
+    """
+    return sorted(_REGISTRY)
+
+
+def _register_namespace(module: typing.Any) -> None:
+    import inspect
+
+    for name, obj in inspect.getmembers(module, inspect.isclass):
+        if name.startswith("_"):
+            continue
+        if obj.__module__ is None or not obj.__module__.startswith(module.__name__):
+            # Skip re-exports from other packages (e.g. numpy classes).
+            continue
+        try:
+            register(obj)
+        except Exception:  # noqa: BLE001
+            pass
+
+
+def _bootstrap() -> None:
+    """Register all ``torch.optim`` and ``lr_scheduler`` classes.
+
+    Called automatically via the ``laco.plugins`` entry point when
+    ``laco-torch`` is installed. Silently does nothing if ``torch`` is not
+    importable.
+    """
+    try:
+        import torch.optim as _optim
+        import torch.optim.lr_scheduler as _sched
+    except ImportError:
+        return
+    _register_namespace(_optim)
+    _register_namespace(_sched)
+
+
+_bootstrap()

laco_torch-1.0.0/sources/laco_torch.egg-info/PKG-INFO

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: laco-torch
+Version: 1.0.0
+Summary: PyTorch optimizer and scheduler configurables for laco.
+Author-email: Kurt Stolle <kurt@khws.io>
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+Requires-Dist: laco>=1.0.0
+Requires-Dist: torch>=2.0
+
+# Laco-Torch
+
+PyTorch optimizer and scheduler configurables for laco.
+
+Part of the [laco](https://github.com/khwstolle/laco) project — see the [root README](../../README.md) for an overview.
+
+## Installation
+
+```bash
+pip install laco-torch
+```
+
+## Features
+
+Auto-registers all `torch.optim` and `lr_scheduler` classes as `@L.configurable`; `register()`, `registered_names()`
+
+## Usage
+
+See [`docs/index.md`](docs/index.md) for the full guide.

laco_torch-1.0.0/sources/laco_torch.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+pyproject.toml
+sources/laco/integrations/torch/__init__.py
+sources/laco/integrations/torch/_core.py
+sources/laco_torch.egg-info/PKG-INFO
+sources/laco_torch.egg-info/SOURCES.txt
+sources/laco_torch.egg-info/dependency_links.txt
+sources/laco_torch.egg-info/entry_points.txt
+sources/laco_torch.egg-info/requires.txt
+sources/laco_torch.egg-info/top_level.txt
+tests/test_laco_torch.py

laco_torch-1.0.0/sources/laco_torch.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

laco_torch-1.0.0/sources/laco_torch.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[laco.plugins]
+torch = laco.integrations.torch:_bootstrap

laco_torch-1.0.0/sources/laco_torch.egg-info/requires.txt

@@ -0,0 +1,2 @@
+laco>=1.0.0
+torch>=2.0

laco_torch-1.0.0/sources/laco_torch.egg-info/top_level.txt

@@ -0,0 +1 @@
+laco

laco_torch-1.0.0/tests/test_laco_torch.py

@@ -0,0 +1,72 @@
+"""Tests for laco-torch."""
+
+import pytest
+
+
+def test_import():
+    import laco.integrations.torch  # noqa: F401
+
+
+def test_public_api():
+    import laco.integrations.torch as laco_torch
+
+    assert hasattr(laco_torch, "register")
+    assert hasattr(laco_torch, "registered_names")
+    assert hasattr(laco_torch, "_bootstrap")
+
+
+def test_registered_names_returns_sorted_list():
+    import laco.integrations.torch as laco_torch
+
+    names = laco_torch.registered_names()
+
+    assert isinstance(names, list)
+    assert names == sorted(names)
+
+
+def test_registered_names_contains_torch_optimizers():
+    pytest.importorskip("torch")
+
+    import laco.integrations.torch as laco_torch
+
+    names = laco_torch.registered_names()
+
+    assert any("Adam" in n for n in names), (
+        f"Expected Adam in registered names, got: {names[:5]}"
+    )
+    assert any("SGD" in n for n in names), (
+        f"Expected SGD in registered names, got: {names[:5]}"
+    )
+
+
+def test_register_is_callable():
+    import laco.integrations.torch as laco_torch
+
+    assert callable(laco_torch.register)
+
+
+def test_register_custom_class():
+    pytest.importorskip("torch")
+
+    import laco.integrations.torch as laco_torch
+    from laco.integrations.torch._core import _REGISTRY
+
+    class _FakeOptimizer:
+        __module__ = "laco.integrations.torch"
+        __qualname__ = "_FakeOptimizer"
+
+    before = set(_REGISTRY.keys())
+    laco_torch.register(_FakeOptimizer)
+    after = set(_REGISTRY.keys())
+
+    assert after - before == {"laco.integrations.torch._FakeOptimizer"}
+
+
+def test_bootstrap_is_idempotent():
+    import laco.integrations.torch as laco_torch
+
+    names_before = set(laco_torch.registered_names())
+    laco_torch._bootstrap()
+    names_after = set(laco_torch.registered_names())
+
+    assert names_after >= names_before