0.0.0+develop

flytekitplugins.wandb.tracking

Directory

Classes

Class Description
wandb_init Abstract class for class decorators.

Variables

Property Type Description
WANDB_CUSTOM_TYPE_VALUE str
WANDB_EXECUTION_TYPE_VALUE str

flytekitplugins.wandb.tracking.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.

Parameter Type
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

Methods

Method Description
execute() This method will be called when the decorated function is called.
get_extra_config() Get the config of the decorator.

execute()

def execute(
    args,
    kwargs,
)

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

Parameter Type
args *args
kwargs **kwargs

get_extra_config()

def get_extra_config()

Get the config of the decorator.