| 1 | """Probe primitives. Each module in this package implements one primitive. |
| 2 | |
| 3 | Importing this package eagerly imports every probe module so their |
| 4 | ``__init_subclass__`` hooks populate the registry. If you're hitting |
| 5 | "unknown probe kind" from :func:`dlm_sway.probes.base.build_probe`, the |
| 6 | fix is to ``import dlm_sway.probes`` before building the probe — which |
| 7 | this ``__init__`` does for you. |
| 8 | """ |
| 9 | |
| 10 | from __future__ import annotations |
| 11 | |
| 12 | # Register every shipped probe with the central registry by importing |
| 13 | # its module. Order is not load-bearing for registration but matches the |
| 14 | # categorical grouping in :mod:`dlm_sway.core.result`. |
| 15 | from dlm_sway.probes import ( # noqa: F401 — imports register the probes |
| 16 | adapter_ablation, |
| 17 | adapter_revert, |
| 18 | calibration_drift, |
| 19 | cluster_kl, |
| 20 | delta_kl, |
| 21 | external_perplexity, |
| 22 | gradient_ghost, |
| 23 | leakage, |
| 24 | multi_turn_coherence, |
| 25 | null_adapter, |
| 26 | paraphrase_invariance, |
| 27 | preference_flip, |
| 28 | prompt_collapse, |
| 29 | section_internalization, |
| 30 | style_fingerprint, |
| 31 | tool_use_fidelity, |
| 32 | training_drift, |
| 33 | ) |