| 1 | """dlm-sway — differential testing for fine-tuned causal language models.""" |
| 2 | |
| 3 | from __future__ import annotations |
| 4 | |
| 5 | from dlm_sway.core.errors import ( |
| 6 | BackendNotAvailableError, |
| 7 | ProbeError, |
| 8 | SpecValidationError, |
| 9 | SwayError, |
| 10 | ) |
| 11 | from dlm_sway.core.model import LoadedModel, Model, ModelSpec |
| 12 | from dlm_sway.core.result import ProbeResult, SuiteResult, SwayScore, Verdict |
| 13 | from dlm_sway.core.scoring import ( |
| 14 | DifferentialBackend, |
| 15 | NullCalibratedBackend, |
| 16 | RollingLogprob, |
| 17 | ScalableDifferentialBackend, |
| 18 | ScoringBackend, |
| 19 | TokenDist, |
| 20 | ) |
| 21 | |
| 22 | __all__ = [ |
| 23 | "BackendNotAvailableError", |
| 24 | "DifferentialBackend", |
| 25 | "LoadedModel", |
| 26 | "Model", |
| 27 | "ModelSpec", |
| 28 | "NullCalibratedBackend", |
| 29 | "ProbeError", |
| 30 | "ProbeResult", |
| 31 | "RollingLogprob", |
| 32 | "ScalableDifferentialBackend", |
| 33 | "ScoringBackend", |
| 34 | "SpecValidationError", |
| 35 | "SuiteResult", |
| 36 | "SwayError", |
| 37 | "SwayScore", |
| 38 | "TokenDist", |
| 39 | "Verdict", |
| 40 | ] |
| 41 | |
| 42 | __version__ = "0.1.0.dev0" |