hydraflow 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

hydraflow/runs.py

@@ -501,6 +501,33 @@ class RunCollection:
         """
         return (func(download_artifacts(run_id=run.info.run_id)) for run in self._runs)
 
+    def group_by(
+        self, names: list[str] | None = None, *args
+    ) -> dict[tuple[str, ...], RunCollection]:
+        """
+        Group the runs by the specified parameter names and return a dictionary
+        where the keys are the parameter values and the values are the runs.
+
+        Args:
+            names (list[str] | None): The parameter names to group by.
+            *args: Additional positional arguments to specify parameter names.
+
+        Returns:
+            A dictionary where the keys are the parameter values and the values
+            are the runs.
+        """
+        names = names[:] if names else []
+        names.extend(args)
+
+        grouped_runs = {}
+        for run in self._runs:
+            key = get_params(run, names)
+            if key not in grouped_runs:
+                grouped_runs[key] = []
+            grouped_runs[key].append(run)
+
+        return {key: RunCollection(runs) for key, runs in grouped_runs.items()}
+
 
 def _param_matches(run: Run, key: str, value: Any) -> bool:
     """
@@ -765,6 +792,13 @@ def try_get_run(runs: list[Run], config: object | None = None, **kwargs) -> Run
     raise ValueError(msg)
 
 
+def get_params(run: Run, names: list[str] | None = None, *args) -> tuple[str, ...]:
+    names = names[:] if names else []
+    names.extend(args)
+
+    return tuple(run.data.params[name] for name in names)
+
+
 def get_param_names(runs: list[Run]) -> list[str]:
     """
     Get the parameter names from the runs.

{hydraflow-0.2.4.dist-info → hydraflow-0.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hydraflow
-Version: 0.2.4
+Version: 0.2.5
 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
@@ -48,7 +48,7 @@ Description-Content-Type: text/markdown
 
 ## Overview
 
-Hydraflow is a powerful library designed to seamlessly integrate
+Hydraflow is a library designed to seamlessly integrate
 [Hydra](https://hydra.cc/) and [MLflow](https://mlflow.org/), making it easier to
 manage and track machine learning experiments. By combining the flexibility of
 Hydra's configuration management with the robust experiment tracking capabilities

{hydraflow-0.2.4.dist-info → hydraflow-0.2.5.dist-info}/RECORD

@@ -3,8 +3,8 @@ hydraflow/asyncio.py,sha256=yh851L315QHzRBwq6r-uwO2oZKgz1JawHp-fswfxT1E,6175
 hydraflow/config.py,sha256=6TCKNQZ3sSrIEvl245T2udwFuknejyN1dMcIVmOHdrQ,2102
 hydraflow/context.py,sha256=8Qn99yCSkCarDDthQ6hjgW80CBBIg0H7fnLvtw4ZXo8,7248
 hydraflow/mlflow.py,sha256=gGr0fvFEllduA-ByHMeEamM39zVY_30tjtEbkSZ4lHA,3659
-hydraflow/runs.py,sha256=0t2xhjV9DMA1CNDzBYrsHiZrDZ6cNsaSTxi0ikf6k8c,29907
-hydraflow-0.2.4.dist-info/METADATA,sha256=Rw8m1Ir6Lio6jja44oPHnSMdlLbK2KtZ46UQRD38Lq8,4148
-hydraflow-0.2.4.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-hydraflow-0.2.4.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.2.4.dist-info/RECORD,,
+hydraflow/runs.py,sha256=41P2aIm7Alem3uKHd-JJdoDzzA4LwrO0rIIZKqZGmdc,31071
+hydraflow-0.2.5.dist-info/METADATA,sha256=KDDgZxTmODbd9fSiwLrURTk7il53CQzGkpGrAshPp1s,4139
+hydraflow-0.2.5.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+hydraflow-0.2.5.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
+hydraflow-0.2.5.dist-info/RECORD,,