pynnlf 0.2.2__py3-none-any.whl

pynnlf/__about__.py

@@ -0,0 +1 @@
+__version__ = "0.2.2"

pynnlf/__init__.py

@@ -0,0 +1,5 @@
+from .__about__ import __version__
+from .api import run_experiment, run_experiment_batch, run_tests
+from .workspace import init
+
+__all__ = ["__version__", "init", "run_experiment", "run_experiment_batch", "run_tests"]

pynnlf/api.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+from pathlib import Path
+from typing import Optional, Literal, Union
+from .runner import run_single, run_batch
+from .tests_runner import run_tests
+
+PathLike = Union[str, Path]
+TestMode = Literal["smoke", "full"]
+
+
+def run_experiment(spec_path: str | Path) -> None:
+    """Run a single experiment from <workspace>/specs/experiment.yaml."""
+    run_single(spec_path)
+
+def run_experiment_batch(spec_path: str | Path) -> None:
+    """Run batch experiments from <workspace>/specs/batch.yaml."""
+    run_batch(spec_path)

pynnlf/discovery.py

@@ -0,0 +1,63 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+from pathlib import Path
+
+def discover_unique_file(directory: Path, prefix: str, suffix: str):
+    """
+    Discover a unique file in a directory by prefix and suffix.
+
+    Example:
+        prefix="m19" suffix=".py" matches: m19_*.py
+        prefix="ds19" suffix=".csv" matches: ds19_*.csv
+
+    Args:
+        directory (Path): Directory to search.
+        prefix (str): Required filename prefix (e.g. "m19" or "ds19").
+        suffix (str): Required suffix (e.g. ".py" or ".csv").
+
+    Returns:
+        Path: The unique matching file path.
+
+    Raises:
+        FileNotFoundError: If no match found.
+        ValueError: If more than one match found.
+    """
+    directory = Path(directory)
+    matches = sorted(directory.glob(f"{prefix}_*{suffix}"))
+    if len(matches) == 0:
+        # also allow exact name without underscore: ds19.csv / m19.py
+        exact = directory / f"{prefix}{suffix}"
+        if exact.exists():
+            return exact
+        raise FileNotFoundError(f"No file found for '{prefix}' in {directory}")
+    if len(matches) > 1:
+        raise ValueError(f"Multiple matches for '{prefix}' in {directory}: {[m.name for m in matches]}")
+    return matches[0]
+
+def discover_model_name(models_dir: Path, model_id: str) -> str:
+    """
+    Discover model file and return model_name (file stem).
+
+    Args:
+        models_dir (Path): <workspace>/models
+        model_id (str): model ID from spec, e.g. "m6" or "m19"
+
+    Returns:
+        str: model_name file stem, e.g. "m6_lr" or "m19_my_model"
+    """
+    p = discover_unique_file(models_dir, model_id, ".py")
+    return p.stem
+
+def discover_dataset_path(data_dir: Path, dataset_id: str) -> Path:
+    """
+    Discover dataset file path from dataset ID.
+
+    Args:
+        data_dir (Path): <workspace>/data
+        dataset_id (str): dataset ID from spec, e.g. "ds0" or "ds19"
+
+    Returns:
+        Path: dataset file path.
+    """
+    return discover_unique_file(data_dir, dataset_id, ".csv")