tenseleyflow/sway / 426e0b3

Browse files

sway(probes): register N2 adapter_ablation + lint fix

Authored by espadonne
SHA
426e0b3627544e45a5c830abc72d367ffcdd4792
Parents
e1af82b
Tree
2b19762

2 changed files

StatusFile+-
M src/dlm_sway/probes/__init__.py 1 0
M tests/unit/test_probe_leakage.py 2 2
src/dlm_sway/probes/__init__.pymodified
@@ -13,6 +13,7 @@ from __future__ import annotations
1313
 # its module. Order is not load-bearing for registration but matches the
1414
 # categorical grouping in :mod:`dlm_sway.core.result`.
1515
 from dlm_sway.probes import (  # noqa: F401 — imports register the probes
16
+    adapter_ablation,
1617
     adapter_revert,
1718
     calibration_drift,
1819
     delta_kl,
tests/unit/test_probe_leakage.pymodified
@@ -37,9 +37,9 @@ class TestFragility:
3737
         assert _fragility(0.0, 0.0) == 0.0
3838
3939
     def test_expected_when_perturbed_dropped(self) -> None:
40
-        import pytest as _pt
40
+        import pytest
4141
42
-        assert _fragility(0.8, 0.2) == _pt.approx(0.75)
42
+        assert _fragility(0.8, 0.2) == pytest.approx(0.75)
4343
4444
4545
 def _prose_section(sid: str, content: str) -> Section: