`ef438ac`

tests/test_cluster_kl_prove_value: same sklearn-free kmeans stub

Authored by mfwolffe <wolffemf@dukes.jmu.edu> 3 weeks ago

Status	File	+	-
M	`tests/unit/test_cluster_kl_prove_value.py`	11	0

tests/unit/test_cluster_kl_prove_value.pymodified

      return _encode
 +def _argmax_kmeans(embeddings: np.ndarray, *, k: int, seed: int) -> np.ndarray:
 +    """sklearn-free stub — cluster by argmax of the one-hot test embeddings."""
 +    del seed
 +    labels = np.argmax(embeddings, axis=1).astype(np.int64)
 +    return labels % k
++
++
  @pytest.fixture
  def monkeyed_embed(monkeypatch: pytest.MonkeyPatch) -> dict[str, np.ndarray]:
      table: dict[str, np.ndarray] = {}
          "dlm_sway.probes.cluster_kl._load_embedder",
          lambda _model_id: _stub_embedder(table),  # type: ignore[arg-type]
+     )
 +    monkeypatch.setattr(
 +        "dlm_sway.probes.cluster_kl._kmeans_cluster",
 +        _argmax_kmeans,
 +    )
      return table