gnn_tracking_hpo.config

Module Contents

Functions

auto_suggest_if_not_fixed(→ Any)	Similar to suggest_if_not_fixed, but automatically chooses the correct
read_json(→ dict[str, Any])	Open and read a json file
get_metadata(*[, test])
get_points_to_evaluate(→ list[dict[str, Any]])	Read json files or read from wandb online and return a list of dicts.
retrieve_config_from_wandb(→ dict[str, Any])	Retrieve configuration of run from wandb based on (part of) a hash

gnn_tracking_hpo.config.auto_suggest_if_not_fixed(key: str, config: dict[str, Any], trial: optuna.Trial, *args, **kwargs) → Any

Similar to suggest_if_not_fixed, but automatically chooses the correct function.

Important: It matters whether the argument types are ints or floats!

gnn_tracking_hpo.config.read_json(path: os.PathLike | str) → dict[str, Any]

Open and read a json file

gnn_tracking_hpo.config.get_metadata(*, test=False)

gnn_tracking_hpo.config.get_points_to_evaluate(paths: None | list[str] | list[os.PathLike] = None) → list[dict[str, Any]]

Read json files or read from wandb online and return a list of dicts. Json files can either contain dictionary (single config) or a list thereof.

gnn_tracking_hpo.config.retrieve_config_from_wandb(hash: str) → dict[str, Any]

Retrieve configuration of run from wandb based on (part of) a hash