WandBLogger

Uses environment variables for authentication:

WANDB_API_KEY: Your WandB API key (required for online logging)
WANDB_ENTITY: Username or team name (optional)
WANDB_PROJECT: Project name (optional, overrides constructor project)

If wandb is not installed, this callback will log warnings but not crash.

Parameters

str

required

WandB project name. Can be overridden by WANDB_PROJECT env var.

dict[str, Any] | None

default:"None"

Hyperparameters and configuration to log.

Methods

on_optimization_start

Initialize WandB run. Parameters

CallbackContext

required

on_step_end

Log metrics to WandB. Parameters

CallbackContext

required

on_optimization_end

Finish WandB run. Parameters

CallbackContext

required

Example

wandb_logger = WandBLogger(
    project="protein-design",
    config={"temperature": 1.0, "n_steps": 1000}
)

EarlyStopping

MonteCarloAnalyzer

Overview

Core

Energies

Minimizers

Mutation

Oracles

Callbacks

Analysis

Parameters

Methods

on_optimization_start

on_step_end

on_optimization_end

Example

​Parameters

​Methods

​on_optimization_start

​on_step_end

​on_optimization_end

​Example

Parameters

Methods

on_optimization_start

on_step_end

on_optimization_end

Example