hydraflow 0.12.3__py3-none-any.whl → 0.12.4__py3-none-any.whl

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.dist-info → hydraflow-0.12.4.dist-info}/METADATA

@@ -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.dist-info → hydraflow-0.12.4.dist-info}/RECORD

@@ -4,7 +4,7 @@ hydraflow/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hydraflow/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hydraflow/core/config.py,sha256=SJzjgsO_kzB78_whJ3lmy7GlZvTvwZONH1BJBn8zCuI,3817
 hydraflow/core/context.py,sha256=L4OygMLbITwlWzq17Lh8VoXKKtjOJ3DBEVsBddKPSJ8,4741
-hydraflow/core/io.py,sha256=7rVEcXus0tES2qZIckIOcBxc9Br-nSEdWzGGA7JedB0,6796
+hydraflow/core/io.py,sha256=6ZQYqxPUkIinFYqSQXJPzTSnuhfP1KfFchTNeNn-g6A,7311
 hydraflow/core/main.py,sha256=dY8uUykS_AbzverrSWkXLyj98TjBPHAiMUf_l5met1U,5162
 hydraflow/core/mlflow.py,sha256=OQJ3f2wkHJRb11ZK__HF4R8FyBEje7-NOqObpoanGhU,5704
 hydraflow/core/param.py,sha256=LHU9j9_7oA99igasoOyKofKClVr9FmGA3UABJ-KmyS0,4538
@@ -17,8 +17,8 @@ hydraflow/executor/conf.py,sha256=2dv6_PlsynRmia-fGZlmBEVt8GopT0f32N13qY7tYnM,40
 hydraflow/executor/io.py,sha256=yZMcBVmAbPZZ82cAXhgiJfj9p8WvHmzOCMBg_vtEVek,1509
 hydraflow/executor/job.py,sha256=IL7ek0Vwa3Bl_gANq0wCbldNCUclo8YBckeEeO6W6xg,4852
 hydraflow/executor/parser.py,sha256=_Rfund3FDgrXitTt_znsTpgEtMDqZ_ICynaB_Zje14Q,14561
-hydraflow-0.12.3.dist-info/METADATA,sha256=Rlwy1HJ4yI5O7bKLSZhZ1JhKIz3tL8Z-TGsREPhfUpw,4549
-hydraflow-0.12.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hydraflow-0.12.3.dist-info/entry_points.txt,sha256=XI0khPbpCIUo9UPqkNEpgh-kqK3Jy8T7L2VCWOdkbSM,48
-hydraflow-0.12.3.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.12.3.dist-info/RECORD,,
+hydraflow-0.12.4.dist-info/METADATA,sha256=CT9bb--1HhyO_tcahfCTzSUBH2nWt7i2lY2pS3Dac38,4549
+hydraflow-0.12.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hydraflow-0.12.4.dist-info/entry_points.txt,sha256=XI0khPbpCIUo9UPqkNEpgh-kqK3Jy8T7L2VCWOdkbSM,48
+hydraflow-0.12.4.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
+hydraflow-0.12.4.dist-info/RECORD,,