Skip to content

Adding Things

Theseus is structured around four extension points. Each one follows the same pattern: write a class, drop it in the right folder, register it.

What Where How
Model theseus/model/models/ subclass Module, add to __init__.py
Experiment theseus/experiments/ @job("key") + subclass BaseTrainer
Dataset theseus/data/datasets/ @dataset("key") + subclass a dataset base
Evaluation theseus/evaluation/datasets/ @evaluation("key") + subclass RolloutEvaluation