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

hydraflow/context.py

@@ -51,9 +51,11 @@ def log_run(
         None
 
     Example:
+        ```python
         with log_run(config):
             # Perform operations within the MLflow run context
             pass
+        ```
     """
     log_params(config, synchronous=synchronous)
 
@@ -122,8 +124,9 @@ def start_run(
             pass
 
     See Also:
-        `mlflow.start_run`: The MLflow function to start a run directly.
-        `log_run`: A context manager to log parameters and manage the MLflow run context.
+        - `mlflow.start_run`: The MLflow function to start a run directly.
+        - `log_run`: A context manager to log parameters and manage the MLflow
+           run context.
     """
     with mlflow.start_run(
         run_id=run_id,
@@ -169,9 +172,11 @@ def watch(
         None
 
     Example:
+        ```python
         with watch(log_artifact, "/path/to/dir"):
             # Perform operations while watching the directory for changes
             pass
+        ```
     """
     dir = dir or get_artifact_dir()
     if isinstance(dir, Path):

hydraflow/mlflow.py

@@ -26,7 +26,7 @@ import joblib
 import mlflow
 from hydra.core.hydra_config import HydraConfig
 from mlflow.entities import ViewType
-from mlflow.tracking.fluent import SEARCH_MAX_RESULTS_PANDAS
+from mlflow.tracking.fluent import SEARCH_MAX_RESULTS_PANDAS, _get_experiment_id
 
 from hydraflow.config import iter_params
 from hydraflow.run_collection import RunCollection
@@ -183,7 +183,8 @@ def list_runs(
         return search_runs(experiment_names=experiment_names)
 
     if experiment_names is None:
-        raise NotImplementedError
+        experiment_id = _get_experiment_id()
+        experiment_names = [mlflow.get_experiment(experiment_id).name]
 
     run_ids = []
 

hydraflow/progress.py

@@ -7,16 +7,12 @@ of tasks in parallel while displaying progress updates.
 
 The following key components are provided:
 
-- JoblibProgress: A context manager for tracking progress with Rich's Progress
+- JoblibProgress: A context manager for tracking progress with Rich's progress
     bar.
 - parallel_progress: A function to execute a given function in parallel over
     an iterable with progress tracking.
 - multi_tasks_progress: A function to render auto-updating progress bars for
     multiple tasks concurrently.
-
-Usage:
-    Import the necessary functions and use them to manage progress in your
-    parallel processing tasks.
 """
 
 from __future__ import annotations
@@ -56,8 +52,10 @@ def JoblibProgress(
         Progress: A Progress instance for managing the progress bar.
 
     Example:
+        ```python
         with JoblibProgress("task", total=100) as progress:
             # Your parallel processing code here
+        ```
     """
     if not columns:
         columns = Progress.get_default_columns()

hydraflow/run_collection.py

@@ -1,31 +1,27 @@
 """
-This module provides functionality for managing and interacting with MLflow runs.
-It includes the `RunCollection` class, which serves as a container for multiple MLflow
-run objects, and various methods to filter, retrieve, and manipulate these runs.
+Provide functionality for managing and interacting with MLflow runs.
+It includes the `RunCollection` class, which serves as a container
+for multiple MLflow `Run` instances, and various methods to filter and
+retrieve these runs.
 
 Key Features:
-- **Run Management**: The `RunCollection` class allows for easy management of multiple
-  MLflow runs, providing methods to access, filter, and sort runs based on various
-  criteria.
-- **Filtering**: The module supports filtering runs based on specific configurations
+- **Run Management**: The `RunCollection` class allows for easy management of
+  multiple MLflow runs, providing methods to filter and retrieve runs based
+  on various criteria.
+- **Filtering**: Support filtering runs based on specific configurations
   and parameters, enabling users to easily find runs that match certain conditions.
 - **Retrieval**: Users can retrieve specific runs, including the first, last, or any
   run that matches a given configuration.
-- **Artifact Handling**: The module provides methods to access and manipulate the
-  artifacts associated with each run, including retrieving artifact URIs and directories.
-
-The `RunCollection` class is designed to work seamlessly with the MLflow tracking
-API, providing a robust solution for managing machine learning experiment runs and
-their associated metadata. This module is particularly useful for data scientists and
-machine learning engineers who need to track and analyze the results of their experiments
-efficiently.
+- **Artifact Handling**: Provide methods to access and manipulate the
+  artifacts associated with each run, including retrieving artifact URIs and
+  directories.
 """
 
 from __future__ import annotations
 
 from dataclasses import dataclass, field
 from itertools import chain
-from typing import TYPE_CHECKING, Any, Concatenate, ParamSpec, TypeVar
+from typing import TYPE_CHECKING, Any, Concatenate, ParamSpec, TypeVar, overload
 
 from mlflow.entities.run import Run
 
@@ -47,9 +43,9 @@ P = ParamSpec("P")
 @dataclass
 class RunCollection:
     """
-    A class to represent a collection of MLflow runs.
+    Represent a collection of MLflow runs.
 
-    This class provides methods to interact with the runs, such as filtering,
+    Provide methods to interact with the runs, such as filtering,
     retrieving specific runs, and accessing run information.
 
     Key Features:
@@ -59,10 +55,10 @@ class RunCollection:
     """
 
     _runs: list[Run]
-    """A list of MLflow Run objects."""
+    """A list of MLflow `Run` instances."""
 
     _info: RunCollectionInfo = field(init=False)
-    """A list of MLflow Run objects."""
+    """An instance of `RunCollectionInfo`."""
 
     def __post_init__(self):
         self._info = RunCollectionInfo(self)
@@ -76,16 +72,51 @@ class RunCollection:
     def __iter__(self) -> Iterator[Run]:
         return iter(self._runs)
 
-    def __getitem__(self, index: int) -> Run:
+    @overload
+    def __getitem__(self, index: int) -> Run: ...
+
+    @overload
+    def __getitem__(self, index: slice) -> RunCollection: ...
+
+    def __getitem__(self, index: int | slice) -> Run | RunCollection:
+        if isinstance(index, slice):
+            return self.__class__(self._runs[index])
+
         return self._runs[index]
 
     def __contains__(self, run: Run) -> bool:
         return run in self._runs
 
+    @classmethod
+    def from_list(cls, runs: list[Run]) -> RunCollection:
+        """Create a new `RunCollection` instance from a list of MLflow `Run` instances."""
+
+        return cls(runs)
+
     @property
     def info(self) -> RunCollectionInfo:
+        """An instance of `RunCollectionInfo`."""
         return self._info
 
+    def take(self, n: int) -> RunCollection:
+        """Take the first n runs from the collection.
+
+        If n is negative, the method returns the last n runs
+        from the collection.
+
+        Args:
+            n (int): The number of runs to take. If n is negative, the method
+            returns the last n runs from the collection.
+
+        Returns:
+            A new `RunCollection` instance containing the first n runs if n is
+            positive, or the last n runs if n is negative.
+        """
+        if n < 0:
+            return self.__class__(self._runs[n:])
+
+        return self.__class__(self._runs[:n])
+
     def sort(
         self,
         key: Callable[[Run], Any] | None = None,
@@ -93,12 +124,37 @@ class RunCollection:
     ) -> None:
         self._runs.sort(key=key or (lambda x: x.info.start_time), reverse=reverse)
 
+    def one(self) -> Run:
+        """
+        Get the only `Run` instance in the collection.
+
+        Returns:
+            The only `Run` instance in the collection.
+
+        Raises:
+            ValueError: If the collection does not contain exactly one run.
+        """
+        if len(self._runs) != 1:
+            raise ValueError("The collection does not contain exactly one run.")
+
+        return self._runs[0]
+
+    def try_one(self) -> Run | None:
+        """
+        Try to get the only `Run` instance in the collection.
+
+        Returns:
+            The only `Run` instance in the collection, or None if the collection
+            does not contain exactly one run.
+        """
+        return self._runs[0] if len(self._runs) == 1 else None
+
     def first(self) -> Run:
         """
-        Get the first run in the collection.
+        Get the first `Run` instance in the collection.
 
         Returns:
-            The first run object in the collection.
+            The first `Run` instance in the collection.
 
         Raises:
             ValueError: If the collection is empty.
@@ -110,20 +166,20 @@ class RunCollection:
 
     def try_first(self) -> Run | None:
         """
-        Try to get the first run in the collection.
+        Try to get the first `Run` instance in the collection.
 
         Returns:
-            The first run object in the collection, or None if the collection
+            The first `Run` instance in the collection, or None if the collection
             is empty.
         """
         return self._runs[0] if self._runs else None
 
     def last(self) -> Run:
         """
-        Get the last run in the collection.
+        Get the last `Run` instance in the collection.
 
         Returns:
-            The last run object in the collection.
+            The last `Run` instance in the collection.
 
         Raises:
             ValueError: If the collection is empty.
@@ -135,17 +191,17 @@ class RunCollection:
 
     def try_last(self) -> Run | None:
         """
-        Try to get the last run in the collection.
+        Try to get the last `Run` instance in the collection.
 
         Returns:
-            The last run object in the collection, or None if the collection is
-            empty.
+            The last `Run` instance in the collection, or None if the collection
+            is empty.
         """
         return self._runs[-1] if self._runs else None
 
     def filter(self, config: object | None = None, **kwargs) -> RunCollection:
         """
-        Filter the runs based on the provided configuration.
+        Filter the `Run` instances based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and additional key-value pairs. The
@@ -173,7 +229,7 @@ class RunCollection:
 
     def find(self, config: object | None = None, **kwargs) -> Run:
         """
-        Find the first run based on the provided configuration.
+        Find the first `Run` instance based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and returns the first run that matches
@@ -185,20 +241,22 @@ class RunCollection:
             **kwargs: Additional key-value pairs to filter the runs.
 
         Returns:
-            The first run object that matches the provided configuration.
+            The first `Run` instance that matches the provided configuration.
 
         Raises:
             ValueError: If no run matches the criteria.
 
         See Also:
-            RunCollection.filter: The method that performs the actual filtering
-            logic.
+            `filter`: Perform the actual filtering logic.
         """
-        return find_run(self._runs, config, **kwargs)
+        try:
+            return self.filter(config, **kwargs).first()
+        except ValueError:
+            raise ValueError("No run matches the provided configuration.")
 
     def try_find(self, config: object | None = None, **kwargs) -> Run | None:
         """
-        Find the first run based on the provided configuration.
+        Try to find the first `Run` instance based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and returns the first run that matches
@@ -210,18 +268,17 @@ class RunCollection:
             **kwargs: Additional key-value pairs to filter the runs.
 
         Returns:
-            The first run object that matches the provided configuration, or
+            The first `Run` instance that matches the provided configuration, or
             None if no runs match the criteria.
 
         See Also:
-            RunCollection.filter: The method that performs the actual filtering
-            logic.
+            `filter`: Perform the actual filtering logic.
         """
-        return try_find_run(self._runs, config, **kwargs)
+        return self.filter(config, **kwargs).try_first()
 
     def find_last(self, config: object | None = None, **kwargs) -> Run:
         """
-        Find the last run based on the provided configuration.
+        Find the last `Run` instance based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and returns the last run that matches
@@ -233,20 +290,22 @@ class RunCollection:
             **kwargs: Additional key-value pairs to filter the runs.
 
         Returns:
-            The last run object that matches the provided configuration.
+            The last `Run` instance that matches the provided configuration.
 
         Raises:
             ValueError: If no run matches the criteria.
 
         See Also:
-            RunCollection.filter: The method that performs the actual filtering
-            logic.
+            `filter`: Perform the actual filtering logic.
         """
-        return find_last_run(self._runs, config, **kwargs)
+        try:
+            return self.filter(config, **kwargs).last()
+        except ValueError:
+            raise ValueError("No run matches the provided configuration.")
 
     def try_find_last(self, config: object | None = None, **kwargs) -> Run | None:
         """
-        Find the last run based on the provided configuration.
+        Try to find the last `Run` instance based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and returns the last run that matches
@@ -258,18 +317,17 @@ class RunCollection:
             **kwargs: Additional key-value pairs to filter the runs.
 
         Returns:
-            The last run object that matches the provided configuration, or
+            The last `Run` instance that matches the provided configuration, or
             None if no runs match the criteria.
 
         See Also:
-            RunCollection.filter: The method that performs the actual filtering
-            logic.
+            `filter`: Perform the actual filtering logic.
         """
-        return try_find_last_run(self._runs, config, **kwargs)
+        return self.filter(config, **kwargs).try_last()
 
     def get(self, config: object | None = None, **kwargs) -> Run:
         """
-        Retrieve a specific run based on the provided configuration.
+        Retrieve a specific `Run` instance based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and returns the run that matches the
@@ -281,21 +339,24 @@ class RunCollection:
             **kwargs: Additional key-value pairs to filter the runs.
 
         Returns:
-            The run object that matches the provided configuration.
+            The `Run` instance that matches the provided configuration.
 
         Raises:
             ValueError: If no run matches the criteria or if more than one run
             matches the criteria.
 
         See Also:
-            RunCollection.filter: The method that performs the actual filtering
-            logic.
+            `filter`: Perform the actual filtering logic.
         """
-        return get_run(self._runs, config, **kwargs)
+        try:
+            return self.filter(config, **kwargs).one()
+        except ValueError:
+            msg = "The filtered collection does not contain exactly one run."
+            raise ValueError(msg)
 
     def try_get(self, config: object | None = None, **kwargs) -> Run | None:
         """
-        Retrieve a specific run based on the provided configuration.
+        Try to retrieve a specific `Run` instance based on the provided configuration.
 
         This method filters the runs in the collection according to the
         specified configuration object and returns the run that matches the
@@ -307,17 +368,16 @@ class RunCollection:
             **kwargs: Additional key-value pairs to filter the runs.
 
         Returns:
-            The run object that matches the provided configuration, or None if
-            no runs match the criteria.
+            The `Run` instance that matches the provided configuration, or None
+            if no runs match the criteria.
 
         Raises:
             ValueError: If more than one run matches the criteria.
 
         See Also:
-            RunCollection.filter: The method that performs the actual filtering
-            logic.
+            `filter`: Perform the actual filtering logic.
         """
-        return try_get_run(self._runs, config, **kwargs)
+        return self.filter(config, **kwargs).try_one()
 
     def get_param_names(self) -> list[str]:
         """
@@ -330,7 +390,13 @@ class RunCollection:
         Returns:
             A list of unique parameter names.
         """
-        return get_param_names(self._runs)
+        param_names = set()
+
+        for run in self:
+            for param in run.data.params.keys():
+                param_names.add(param)
+
+        return list(param_names)
 
     def get_param_dict(self) -> dict[str, list[str]]:
         """
@@ -345,7 +411,13 @@ class RunCollection:
             A dictionary where the keys are parameter names and the values are
             lists of parameter values.
         """
-        return get_param_dict(self._runs)
+        params = {}
+
+        for name in self.get_param_names():
+            it = (run.data.params[name] for run in self if name in run.data.params)
+            params[name] = sorted(set(it))
+
+        return params
 
     def map(
         self,
@@ -426,9 +498,9 @@ class RunCollection:
         Apply a function to each artifact URI in the collection and return an
         iterator of results.
 
-        This method iterates over each run in the collection, retrieves the
-        artifact URI, and applies the provided function to it. If a run does not
-        have an artifact URI, None is passed to the function.
+        Iterate over each run in the collection, retrieves the artifact URI, and
+        apply the provided function to it. If a run does not have an artifact
+        URI, None is passed to the function.
 
         Args:
             func (Callable[[str | None, P], T]): A function that takes an
@@ -452,9 +524,8 @@ class RunCollection:
         Apply a function to each artifact directory in the collection and return
         an iterator of results.
 
-        This method iterates over each run in the collection, downloads the
-        artifact directory, and applies the provided function to the directory
-        path.
+        Iterate over each run in the collection, downloads the artifact
+        directory, and apply the provided function to the directory path.
 
         Args:
             func (Callable[[Path, P], T]): A function that takes an artifact directory
@@ -474,7 +545,7 @@ class RunCollection:
         """
         Group runs by specified parameter names.
 
-        This method groups the runs in the collection based on the values of the
+        Group the runs in the collection based on the values of the
         specified parameters. Each unique combination of parameter values will
         form a key in the returned dictionary.
 
@@ -500,7 +571,7 @@ def _param_matches(run: Run, key: str, value: Any) -> bool:
     """
     Check if the run's parameter matches the specified key-value pair.
 
-    This function checks if the run's parameters contain the specified
+    Check if the run's parameters contain the specified
     key-value pair. It handles different types of values, including lists
     and tuples.
 
@@ -530,11 +601,17 @@ 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.
 
-    This method filters the runs in the collection according to the
+    Filter the runs in the collection according to the
     specified configuration object and additional key-value pairs.
     The configuration object and key-value pairs should contain
     key-value pairs that correspond to the parameters of the runs.
@@ -552,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:
@@ -563,202 +641,16 @@ def filter_runs(runs: list[Run], config: object | None = None, **kwargs) -> list
         if len(runs) == 0:
             return []
 
-    return runs
-
-
-def find_run(runs: list[Run], config: object | None = None, **kwargs) -> Run:
-    """
-    Find the first run based on the provided configuration.
-
-    This method filters the runs in the collection according to the
-    specified configuration object and returns the first run that matches
-    the provided parameters. If no run matches the criteria, a `ValueError` is
-    raised.
-
-    Args:
-        runs (list[Run]): The runs to filter.
-        config (object | None): The configuration object to identify the run.
-        **kwargs: Additional key-value pairs to filter the runs.
-
-    Returns:
-        The first run object that matches the provided configuration.
-
-    Raises:
-        ValueError: If no run matches the criteria.
-
-    See Also:
-        RunCollection.filter: The method that performs the actual filtering logic.
-    """
-    filtered_runs = filter_runs(runs, config, **kwargs)
-
-    if len(filtered_runs) == 0:
-        raise ValueError("No run matches the provided configuration.")
-
-    return filtered_runs[0]
-
-
-def try_find_run(runs: list[Run], config: object | None = None, **kwargs) -> Run | None:
-    """
-    Find the first run based on the provided configuration.
-
-    This method filters the runs in the collection according to the
-    specified configuration object and returns the first run that matches
-    the provided parameters. If no run matches the criteria, None is returned.
-
-    Args:
-        runs (list[Run]): The runs to filter.
-        config (object | None): The configuration object to identify the run.
-        **kwargs: Additional key-value pairs to filter the runs.
-
-    Returns:
-        The first run object that matches the provided configuration, or None
-        if no runs match the criteria.
-    """
-    filtered_runs = filter_runs(runs, config, **kwargs)
-
-    if len(filtered_runs) == 0:
-        return None
-
-    return filtered_runs[0]
-
-
-def find_last_run(runs: list[Run], config: object | None = None, **kwargs) -> Run:
-    """
-    Find the last run based on the provided configuration.
-
-    This method filters the runs in the collection according to the
-    specified configuration object and returns the last run that matches
-    the provided parameters. If no run matches the criteria, a `ValueError`
-    is raised.
-
-    Args:
-        runs (list[Run]): The runs to filter.
-        config (object | None): The configuration object to identify the run.
-        **kwargs: Additional key-value pairs to filter the runs.
-
-    Returns:
-        The last run object that matches the provided configuration.
-
-    Raises:
-        ValueError: If no run matches the criteria.
-
-    See Also:
-        RunCollection.filter: The method that performs the actual filtering
-        logic.
-    """
-    filtered_runs = filter_runs(runs, config, **kwargs)
-
-    if len(filtered_runs) == 0:
-        raise ValueError("No run matches the provided configuration.")
-
-    return filtered_runs[-1]
-
-
-def try_find_last_run(
-    runs: list[Run], config: object | None = None, **kwargs
-) -> Run | None:
-    """
-    Find the last run based on the provided configuration.
-
-    This method filters the runs in the collection according to the
-    specified configuration object and returns the last run that matches
-    the provided parameters. If no run matches the criteria, None is returned.
-
-    Args:
-        runs (list[Run]): The runs to filter.
-        config (object | None): The configuration object to identify the run.
-        **kwargs: Additional key-value pairs to filter the runs.
-
-    Returns:
-        The last run object that matches the provided configuration, or None
-        if no runs match the criteria.
-    """
-    filtered_runs = filter_runs(runs, config, **kwargs)
-
-    if len(filtered_runs) == 0:
-        return None
-
-    return filtered_runs[-1]
-
-
-def get_run(runs: list[Run], config: object | None = None, **kwargs) -> Run:
-    """
-    Retrieve a specific run based on the provided configuration.
-
-    This method filters the runs in the collection according to the
-    specified configuration object and returns the run that matches
-    the provided parameters. If no run matches the criteria, or if more
-    than one run matches the criteria, a `ValueError` is raised.
-
-    Args:
-        runs (list[Run]): The runs to filter.
-        config (object | None): The configuration object to identify the run.
-        **kwargs: Additional key-value pairs to filter the runs.
-
-    Returns:
-        The run object that matches the provided configuration.
-
-    Raises:
-        ValueError: If no run matches the criteria or if more than one run
-        matches the criteria.
-
-    See Also:
-        RunCollection.filter: The method that performs the actual filtering
-        logic.
-    """
-    filtered_runs = filter_runs(runs, config, **kwargs)
-
-    if len(filtered_runs) == 0:
-        raise ValueError("No run matches the provided configuration.")
-
-    if len(filtered_runs) == 1:
-        return filtered_runs[0]
-
-    msg = (
-        f"Multiple runs were filtered. Expected number of runs is 1, "
-        f"but found {len(filtered_runs)} runs."
-    )
-    raise ValueError(msg)
-
-
-def try_get_run(runs: list[Run], config: object | None = None, **kwargs) -> Run | None:
-    """
-    Retrieve a specific run based on the provided configuration.
-
-    This method filters the runs in the collection according to the
-    specified configuration object and returns the run that matches
-    the provided parameters. If no run matches the criteria, None is returned.
-    If more than one run matches the criteria, a `ValueError` is raised.
-
-    Args:
-        runs (list[Run]): The runs to filter.
-        config (object | None): The configuration object to identify the run.
-        **kwargs: Additional key-value pairs to filter the runs.
+    if isinstance(status, str) and status.startswith("!"):
+        status = status[1:].lower()
+        return [run for run in runs if run.info.status.lower() != status]
 
-    Returns:
-        The run object that matches the provided configuration, or None
-        if no runs match the criteria.
-
-    Raises:
-        ValueError: If more than one run matches the criteria.
-
-    See Also:
-        RunCollection.filter: The method that performs the actual filtering
-        logic.
-    """
-    filtered_runs = filter_runs(runs, config, **kwargs)
+    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]
 
-    if len(filtered_runs) == 0:
-        return None
-
-    if len(filtered_runs) == 1:
-        return filtered_runs[0]
-
-    msg = (
-        "Multiple runs were filtered. Expected number of runs is 1, "
-        f"but found {len(filtered_runs)} runs."
-    )
-    raise ValueError(msg)
+    return runs
 
 
 def get_params(run: Run, *names: str | list[str]) -> tuple[str | None, ...]:
@@ -787,52 +679,3 @@ def get_params(run: Run, *names: str | list[str]) -> tuple[str | None, ...]:
             names_.append(name)
 
     return tuple(run.data.params.get(name) for name in names_)
-
-
-def get_param_names(runs: list[Run]) -> list[str]:
-    """
-    Get the parameter names from the runs.
-
-    This method extracts the unique parameter names from the provided list of
-    runs. It iterates through each run and collects the parameter names into a
-    set to ensure uniqueness.
-
-    Args:
-        runs (list[Run]): The list of runs from which to extract parameter names.
-
-    Returns:
-        A list of unique parameter names.
-    """
-    param_names = set()
-
-    for run in runs:
-        for param in run.data.params.keys():
-            param_names.add(param)
-
-    return list(param_names)
-
-
-def get_param_dict(runs: list[Run]) -> dict[str, list[str]]:
-    """
-    Get the parameter dictionary from the list of runs.
-
-    This method extracts the parameter names and their corresponding values
-    from the provided list of runs. It iterates through each run and collects
-    the parameter values into a dictionary where the keys are parameter names
-    and the values are lists of parameter values.
-
-    Args:
-        runs (list[Run]): The list of runs from which to extract parameter names
-        and values.
-
-    Returns:
-        A dictionary where the keys are parameter names and the values are lists
-        of parameter values.
-    """
-    params = {}
-
-    for name in get_param_names(runs):
-        it = (run.data.params[name] for run in runs if name in run.data.params)
-        params[name] = sorted(set(it))
-
-    return params

{hydraflow-0.2.13.dist-info → hydraflow-0.2.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: hydraflow
-Version: 0.2.13
+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.15.dist-info/RECORD

@@ -0,0 +1,12 @@
+hydraflow/__init__.py,sha256=B7rWSiGP5WwWjijcb41Bv9uuo5MQ6gbBbVWGAWYtK-k,598
+hydraflow/asyncio.py,sha256=jdXuEFC6f7L_Dq6beASFZPQSvCnGimVxU-PRFsNc5U0,6241
+hydraflow/config.py,sha256=6TCKNQZ3sSrIEvl245T2udwFuknejyN1dMcIVmOHdrQ,2102
+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=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,,

hydraflow-0.2.13.dist-info/RECORD

@@ -1,12 +0,0 @@
-hydraflow/__init__.py,sha256=B7rWSiGP5WwWjijcb41Bv9uuo5MQ6gbBbVWGAWYtK-k,598
-hydraflow/asyncio.py,sha256=jdXuEFC6f7L_Dq6beASFZPQSvCnGimVxU-PRFsNc5U0,6241
-hydraflow/config.py,sha256=6TCKNQZ3sSrIEvl245T2udwFuknejyN1dMcIVmOHdrQ,2102
-hydraflow/context.py,sha256=G7JMrG70sgBH2qILXl5nkGWNUoRggj518JWUq0ZiJ9E,7776
-hydraflow/info.py,sha256=Vj2sT66Ric63mmaq7Yu8nDFhsGQYO3MCHrxFpapDufc,3458
-hydraflow/mlflow.py,sha256=QlJlhsOAJ3t9hcvQRUgeE5od6g1AbyFNYQIxkq1FCoM,8140
-hydraflow/progress.py,sha256=uaxS9UnwsikOPB_Ho-5mXdBYz3Zedj-SL9FzT094wjY,6623
-hydraflow/run_collection.py,sha256=V5lGdGHYgsSpBOYGaVEL1mpKJvdiEshBL0KmmZ8qeZo,29161
-hydraflow-0.2.13.dist-info/METADATA,sha256=DRWv1h7QW68HH7BGkyiMbIha1d-VlV95eO-X4PRj44U,4088
-hydraflow-0.2.13.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-hydraflow-0.2.13.dist-info/licenses/LICENSE,sha256=IGdDrBPqz1O0v_UwCW-NJlbX9Hy9b3uJ11t28y2srmY,1062
-hydraflow-0.2.13.dist-info/RECORD,,