Allow direct importation of QuadraticEditor

EleutherAI · Jan 10, 2024 · 42adfae · 42adfae
1 parent 644ec94
commit 42adfae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/concept_erasure/__init__.py b/concept_erasure/__init__.py
@@ -2,7 +2,7 @@
 from .groupby import GroupedTensor, groupby
 from .leace import ErasureMethod, LeaceEraser, LeaceFitter
 from .oracle import OracleEraser, OracleFitter
-from .quadratic import QuadraticEraser, QuadraticFitter
+from .quadratic import QuadraticEditor, QuadraticEraser, QuadraticFitter
 from .shrinkage import optimal_linear_shrinkage
 from .utils import assert_type
 
@@ -17,6 +17,7 @@
     "LeaceFitter",
     "OracleEraser",
     "OracleFitter",
+    "QuadraticEditor",
     "QuadraticEraser",
     "QuadraticFitter",
 ]