Small code fix for differential_privacy.py

advaithsrao · Dec 5, 2023 · 5c3d5bb · 5c3d5bb
1 parent 18fe3f3
commit 5c3d5bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ethics/differential_privacy.py b/ethics/differential_privacy.py
@@ -26,7 +26,7 @@
 from mlflow.sklearn import save_model
 from scipy.sparse import hstack
 
-from base import BaseDistilbertModel
+from ethics.base import BaseDistilbertModel
 from utils.util_modeler import Word2VecEmbedder, TPSampler
 
 from opacus import PrivacyEngine