hydraflow 0.2.14__tar.gz → 0.2.15__tar.gz

{hydraflow-0.2.14 → hydraflow-0.2.15}/PKG-INFO

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

{hydraflow-0.2.14 → hydraflow-0.2.15}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "hydraflow"
-version = "0.2.14"
+version = "0.2.15"
 description = "Hydraflow integrates Hydra and MLflow to manage and track machine learning experiments."
 readme = "README.md"
 license = "MIT"

{hydraflow-0.2.14 → hydraflow-0.2.15}/src/hydraflow/run_collection.py

@@ -601,7 +601,13 @@ def _param_matches(run: Run, key: str, value: Any) -> bool:
     return type(value)(param) == value
 
 
-def filter_runs(runs: list[Run], config: object | None = None, **kwargs) -> list[Run]:
+def filter_runs(
+    runs: list[Run],
+    config: object | None = None,
+    *,
+    status: str | list[str] | None = None,
+    **kwargs,
+) -> list[Run]:
     """
     Filter the runs based on the provided configuration.
 
@@ -623,6 +629,7 @@ def filter_runs(runs: list[Run], config: object | None = None, **kwargs) -> list
         config (object | None): The configuration object to filter the runs.
             This can be any object that provides key-value pairs through the
             `iter_params` function.
+        status (str | list[str] | None): The status of the runs to filter.
         **kwargs: Additional key-value pairs to filter the runs.
 
     Returns:
@@ -634,6 +641,15 @@ def filter_runs(runs: list[Run], config: object | None = None, **kwargs) -> list
         if len(runs) == 0:
             return []
 
+    if isinstance(status, str) and status.startswith("!"):
+        status = status[1:].lower()
+        return [run for run in runs if run.info.status.lower() != status]
+
+    if status:
+        status = [status] if isinstance(status, str) else status
+        status = [s.lower() for s in status]
+        return [run for run in runs if run.info.status.lower() in status]
+
     return runs
 
 

{hydraflow-0.2.14 → hydraflow-0.2.15}/tests/test_run_collection.py

@@ -90,6 +90,16 @@ def test_filter_invalid_param(run_list: list[Run]):
     assert len(x) == 6
 
 
+def test_filter_status(run_list: list[Run]):
+    from hydraflow.run_collection import filter_runs
+
+    assert not filter_runs(run_list, status="RUNNING")
+    assert filter_runs(run_list, status="finished") == run_list
+    assert filter_runs(run_list, status=["finished", "running"]) == run_list
+    assert filter_runs(run_list, status="!RUNNING") == run_list
+    assert not filter_runs(run_list, status="!finished")
+
+
 def test_get_params(run_list: list[Run]):
     from hydraflow.run_collection import get_params