hydraflow 0.2.14__py3-none-any.whl → 0.2.15__py3-none-any.whl

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

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

@@ -5,8 +5,8 @@ hydraflow/context.py,sha256=kNJgnDFd_dFoHMMdGPYLbdtvu0p1BgVhA7jQ7bc0m8s,7851
 hydraflow/info.py,sha256=Vj2sT66Ric63mmaq7Yu8nDFhsGQYO3MCHrxFpapDufc,3458
 hydraflow/mlflow.py,sha256=8NJPjWbpedjJ9ylaS4a_NZJ2psmsSlYjv69i6Zth1yM,8242
 hydraflow/progress.py,sha256=_9hrCv2mQFNs3dnsbhBsxrXnzpBdtA2TmfFBVNNLqV4,6539
-hydraflow/run_collection.py,sha256=V6522xC9QvNI5Cwy6dAD5Y7qKhfI6kxByc4vHITuQUs,23293
-hydraflow-0.2.14.dist-info/METADATA,sha256=HnTBaONNqaET5b9m6n1GHt4avPQqaTemcuDXGuldZE4,4088
-hydraflow-0.2.14.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-hydraflow-0.2.14.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.2.14.dist-info/RECORD,,
+hydraflow/run_collection.py,sha256=gJmHooFe-8RuRiNEoFGoKjX0-m7Doqu6mZ1nrrolZCo,23803
+hydraflow-0.2.15.dist-info/METADATA,sha256=OEcnQW4OaPbT78UdXyhWLdfMeA83SETw27UZbXu1VTE,4088
+hydraflow-0.2.15.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+hydraflow-0.2.15.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
+hydraflow-0.2.15.dist-info/RECORD,,