wandb_osh.ray_hooks

Module Contents

Classes

TriggerWandbSyncRayHook

Hook to be used when interfacing wandb with ray tune.

class wandb_osh.ray_hooks.TriggerWandbSyncRayHook(communication_dir: os.PathLike = _comm_default_dir)

Bases: ray.tune.logger.LoggerCallback

Hook to be used when interfacing wandb with ray tune.

Parameters:

communication_dir – Directory used for communication with wandb-osh.

log_trial_result(iteration: int, trial: ray.tune.experiment.trial.Trial, result: dict)