Skip to content

Read results

Utility functions for reading and processing results.

process_seed #

process_seed(
    *,
    path: str | Path,
    seed: str | int | None,
    key_to_extract: str | None = None,
    consider_continuations: bool = False,
    n_workers: int = 1
) -> tuple[list[float], list[float], float]

Reads and processes data per seed.

Source code in neps\plot\read_results.py 
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
def process_seed(
    *,
    path: str | Path,
    seed: str | int | None,
    key_to_extract: str | None = None,  # noqa: ARG001
    consider_continuations: bool = False,  # noqa: ARG001
    n_workers: int = 1,  # noqa: ARG001
) -> tuple[list[float], list[float], float]:
    """Reads and processes data per seed."""
    path = Path(path)
    if seed is not None:
        path = path / str(seed) / "neps_root_directory"

    _fulldf, _summary = neps.status(path, print_summary=False)
    raise NotImplementedError(
        "I'm sorry, I broke this. We now dump all the information neps has available"
        " into the above dataframe `fulldf`."
    )