set

etowahadams · Oct 16, 2024 · 367a638 · 367a638
1 parent 9b35eb5
commit 367a638
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/plm_interpretability/logistic_regression_probe/all_latents.py b/plm_interpretability/logistic_regression_probe/all_latents.py
@@ -100,6 +100,7 @@ def all_latents(
                 plm_model=plm_model,
                 sae_model=sae_model,
                 plm_layer=plm_layer,
+                pool_over_annotation=False,
             )
             with warnings.catch_warnings():
                 # LogisticRegression throws warnings when it can't converge.