Add

etowahadams · Oct 14, 2024 · 3a6310c · 3a6310c
1 parent 220454f
commit 3a6310c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,7 +20,7 @@ requires-python = ">=3.10"
 readme = "README.md"
 
 [tool.setuptools]
-packages = ["plm_interpretability", "plm_interpretability.autointerp"]
+packages = ["plm_interpretability", "plm_interpretability.autointerp", "plm_interpretability.latent_probe"]
 
 [project.scripts]
 autointerp = "plm_interpretability.autointerp.__main__:cli"