From 42adfaed3ca85f38b60831e5dded216a1adb02f9 Mon Sep 17 00:00:00 2001 From: Nora Belrose Date: Wed, 10 Jan 2024 19:42:17 +0000 Subject: [PATCH] Allow direct importation of QuadraticEditor --- concept_erasure/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/concept_erasure/__init__.py b/concept_erasure/__init__.py index f1793c1..e0ffe8b 100644 --- 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", ]