| 1 | """sway — differential testing for fine-tuned causal language models. |
| 2 | |
| 3 | Published on PyPI as ``dlm-sway`` (the short name is taken); the CLI |
| 4 | entry point and source repo are ``sway``. |
| 5 | """ |
| 6 | |
| 7 | from __future__ import annotations |
| 8 | |
| 9 | from dlm_sway.core.errors import ( |
| 10 | BackendNotAvailableError, |
| 11 | ProbeError, |
| 12 | SpecValidationError, |
| 13 | SwayError, |
| 14 | ) |
| 15 | from dlm_sway.core.model import LoadedModel, Model, ModelSpec |
| 16 | from dlm_sway.core.result import ProbeResult, SuiteResult, SwayScore, Verdict, safe_finalize |
| 17 | from dlm_sway.core.scoring import ( |
| 18 | DifferentialBackend, |
| 19 | NullCalibratedBackend, |
| 20 | PreflightCheckable, |
| 21 | RollingLogprob, |
| 22 | ScalableDifferentialBackend, |
| 23 | ScoringBackend, |
| 24 | TokenDist, |
| 25 | ) |
| 26 | |
| 27 | __all__ = [ |
| 28 | "BackendNotAvailableError", |
| 29 | "DifferentialBackend", |
| 30 | "LoadedModel", |
| 31 | "Model", |
| 32 | "ModelSpec", |
| 33 | "NullCalibratedBackend", |
| 34 | "PreflightCheckable", |
| 35 | "ProbeError", |
| 36 | "ProbeResult", |
| 37 | "RollingLogprob", |
| 38 | "ScalableDifferentialBackend", |
| 39 | "ScoringBackend", |
| 40 | "SpecValidationError", |
| 41 | "SuiteResult", |
| 42 | "SwayError", |
| 43 | "SwayScore", |
| 44 | "TokenDist", |
| 45 | "Verdict", |
| 46 | "safe_finalize", |
| 47 | ] |
| 48 | |
| 49 | __version__ = "0.1.0" |