2.0.0b53

WandbSweep

Package: flyteplugins.wandb

Generates a Weights & Biases Sweep link.

class WandbSweep(
    host: str,
    project: typing.Optional[str],
    entity: typing.Optional[str],
    id: typing.Optional[str],
    name: str,
)
Parameter Type Description
host str Base W&B host URL
project typing.Optional[str] W&B project name (overrides context config if provided)
entity typing.Optional[str] W&B entity/team name (overrides context config if provided)
id typing.Optional[str] Optional W&B sweep ID (overrides context config if provided)
name str Link name in the Flyte UI

Methods

Method Description
get_link() Returns a task log link given the action.
def get_link(
    run_name: str,
    project: str,
    domain: str,
    context: typing.Dict[str, str],
    parent_action_name: str,
    action_name: str,
    pod_name: str,
    kwargs,
) -> str

Returns a task log link given the action. Link can have template variables that are replaced by the backend.

Parameter Type Description
run_name str The name of the run.
project str The project name.
domain str The domain name.
context typing.Dict[str, str] Additional context for generating the link.
parent_action_name str The name of the parent action.
action_name str The name of the action.
pod_name str The name of the pod.
kwargs **kwargs Additional keyword arguments. :return: The generated link.