hydraflow 0.12.3__tar.gz → 0.12.4__tar.gz

hydraflow-0.12.4/.github/workflows/publish.yaml

@@ -0,0 +1,38 @@
+name: Publish
+
+on:
+  push:
+    tags:
+      - "[0-9]+.[0-9]+.[0-9]+"
+
+jobs:
+  publish:
+    name: Publish
+    runs-on: ubuntu-latest
+    permissions:
+      id-token: write
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Verify version match
+        run: |
+          TAG=${GITHUB_REF#refs/tags/}
+          PYPROJECT_VERSION=$(grep -m 1 "version = " pyproject.toml | cut -d'"' -f2)
+          if [ "$TAG" != "$PYPROJECT_VERSION" ]; then
+            echo "Error: Git tag ($TAG) does not match pyproject.toml version ($PYPROJECT_VERSION)"
+            exit 1
+          fi
+          echo "Version verified: $TAG"
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: 3.13
+      - name: Install uv
+        run: pip install uv
+      - name: Install the project
+        run: uv sync
+      - name: Build the project
+        run: uv build --no-sources
+      - name: Upload the project to PyPI
+        run: uv publish

{hydraflow-0.12.3 → hydraflow-0.12.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hydraflow
-Version: 0.12.3
+Version: 0.12.4
 Summary: Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments.
 Project-URL: Documentation, https://daizutabi.github.io/hydraflow/
 Project-URL: Source, https://github.com/daizutabi/hydraflow

{hydraflow-0.12.3 → hydraflow-0.12.4}/docs/index.md

@@ -1,10 +1,10 @@
-# Hydraflow Documentation
-
-Hydraflow integrates [Hydra](https://hydra.cc/) and [MLflow](https://mlflow.org/)
-to manage and track machine learning experiments.
-
-## Installation
-
-```bash
-pip install hydraflow
+# Hydraflow Documentation
+
+Hydraflow integrates [Hydra](https://hydra.cc/) and [MLflow](https://mlflow.org/)
+to manage and track machine learning experiments.
+
+## Installation
+
+```bash
+pip install hydraflow
 ```

{hydraflow-0.12.3 → hydraflow-0.12.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hydraflow"
-version = "0.12.3"
+version = "0.12.4"
 description = "Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments."
 readme = "README.md"
 license = { file = "LICENSE" }

{hydraflow-0.12.3 → hydraflow-0.12.4}/src/hydraflow/core/io.py

@@ -12,7 +12,7 @@ from hydra.core.hydra_config import HydraConfig
 from omegaconf import DictConfig, ListConfig, OmegaConf
 
 if TYPE_CHECKING:
-    from collections.abc import Iterable, Iterator
+    from collections.abc import Callable, Iterable, Iterator
 
     from mlflow.entities import Run
 
@@ -179,8 +179,29 @@ def get_experiment_name(path: Path) -> str | None:
     return None
 
 
+def predicate_experiment_dir(
+    path: Path,
+    experiment_names: list[str] | Callable[[str], bool] | None = None,
+) -> bool:
+    """Predicate an experiment directory based on the path and experiment names."""
+    if not path.is_dir() or path.name in [".trash", "0"]:
+        return False
+
+    name = get_experiment_name(path)
+    if not name:
+        return False
+
+    if experiment_names is None:
+        return True
+
+    if isinstance(experiment_names, list):
+        return name in experiment_names
+
+    return experiment_names(name)
+
+
 def iter_experiment_dirs(
-    experiment_names: str | list[str] | None = None,
+    experiment_names: str | list[str] | Callable[[str], bool] | None = None,
     root_dir: str | Path | None = None,
 ) -> Iterator[Path]:
     """Iterate over the experiment directories in the root directory."""
@@ -189,14 +210,12 @@ def iter_experiment_dirs(
 
     root_dir = get_root_dir(root_dir)
     for path in root_dir.iterdir():
-        if path.is_dir() and path.name not in [".trash", "0"]:
-            if name := get_experiment_name(path):
-                if experiment_names is None or name in experiment_names:
-                    yield path
+        if predicate_experiment_dir(path, experiment_names):
+            yield path
 
 
 def iter_run_dirs(
-    experiment_names: str | list[str] | None = None,
+    experiment_names: str | list[str] | Callable[[str], bool] | None = None,
     root_dir: str | Path | None = None,
 ) -> Iterator[Path]:
     """Iterate over the run directories in the root directory."""
@@ -207,7 +226,7 @@ def iter_run_dirs(
 
 
 def iter_artifacts_dirs(
-    experiment_names: str | list[str] | None = None,
+    experiment_names: str | list[str] | Callable[[str], bool] | None = None,
     root_dir: str | Path | None = None,
 ) -> Iterator[Path]:
     """Iterate over the artifacts directories in the root directory."""
@@ -217,7 +236,7 @@ def iter_artifacts_dirs(
 
 def iter_artifact_paths(
     artifact_path: str | Path,
-    experiment_names: str | list[str] | None = None,
+    experiment_names: str | list[str] | Callable[[str], bool] | None = None,
     root_dir: str | Path | None = None,
 ) -> Iterator[Path]:
     """Iterate over the artifact paths in the root directory."""

{hydraflow-0.12.3 → hydraflow-0.12.4}/tests/core/io/test_iter_dirs.py

@@ -49,6 +49,19 @@ def test_iter_experiment_dirs_filter():
     assert [get_experiment_name(p) for p in it] == ["e1"]
 
 
+def test_iter_experiment_dirs_filter_callable():
+    from hydraflow.core.io import get_experiment_name, iter_experiment_dirs
+
+    it = iter_experiment_dirs(lambda name: name == "e2")
+    assert [get_experiment_name(p) for p in it] == ["e2"]
+
+
+def test_predicate_experiment_dir():
+    from hydraflow.core.io import predicate_experiment_dir
+
+    assert predicate_experiment_dir(Path()) is False
+
+
 def test_get_experiment_name_none(root_dir: Path):
     from hydraflow.core.io import get_experiment_name