noether.core.trackers.wandb_tracker

Attributes

WANDB_IMPORT_ERROR

Classes

WandBTrackerSchema

Base configuration for experiment trackers. All tracker configs should inherit from this class.

WandBTracker

Weights and Biases tracker.

Module Contents

noether.core.trackers.wandb_tracker.WANDB_IMPORT_ERROR = None
class noether.core.trackers.wandb_tracker.WandBTrackerSchema(/, **data)

Bases: noether.core.trackers.BaseTrackerConfig

Base configuration for experiment trackers. All tracker configs should inherit from this class.

Create a new model by parsing and validating input data from keyword arguments.

Raises [ValidationError][pydantic_core.ValidationError] if the input data cannot be validated to form a valid model.

self is explicitly positional-only to allow self as a field name.

Parameters:

data (Any)

entity: str | None = None

The entity name for the W&B project.

project: str | None = None

The project name for the W&B project.

mode: Literal['disabled', 'online', 'offline'] | None = None

Tracking mode. Can be ‘disabled’, ‘online’, or ‘offline’.

tags: list[str] | None = None

Optional tags for the W&B run.

class noether.core.trackers.wandb_tracker.WandBTracker(tracker_config, **kwargs)

Bases: noether.core.trackers.BaseTracker

Weights and Biases tracker.

Example config:

- kind: noether.core.trackers.WandBTracker
  mode: online
  entity: my_entity
  project: my_project
Parameters:

  • tracker_config (WandBTrackerSchema) – Configuration for the WandBTracker. See WandBTrackerSchema for available options including mode, entity, and project.

  • **kwargs – Additional keyword arguments passed to the parent class.

MODES = ['disabled', 'online', 'offline']
mode
entity
project
tags