From 367a63892704e84292d08e300970908ee6dd3cc6 Mon Sep 17 00:00:00 2001
From: Liam <liambai2000@gmail.com>
Date: Wed, 16 Oct 2024 17:41:50 -0400
Subject: [PATCH] set

---
 plm_interpretability/logistic_regression_probe/all_latents.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/plm_interpretability/logistic_regression_probe/all_latents.py b/plm_interpretability/logistic_regression_probe/all_latents.py
index d20c6bc..dace4dc 100644
--- 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.