flytekitplugins.wandb

.. currentmodule:: flytekitplugins.wandb

This package contains things that are useful when extending Flytekit.

.. autosummary:: :template: custom.rst :toctree: generated/

wandb_init

Directory

Classes

flytekitplugins.wandb.wandb_init

Abstract class for class decorators. We can attach config on the decorator class and use it in the upper level.

class wandb_init(
    task_function: typing.Optional[typing.Callable],
    project: typing.Optional[str],
    entity: typing.Optional[str],
    secret: typing.Union[flytekit.models.security.Secret, typing.Callable, NoneType],
    id: typing.Optional[str],
    host: str,
    api_host: str,
    init_kwargs: dict,
)

Weights and Biases plugin.

Methods

execute()

def execute(
    args,
    kwargs,
)

This method will be called when the decorated function is called.

get_extra_config()

def get_extra_config()

Get the config of the decorator.