From 5c3d5bb3f340776e36b32314655e438ed31690f5 Mon Sep 17 00:00:00 2001 From: Advaith Rao Date: Mon, 4 Dec 2023 19:44:17 -0500 Subject: [PATCH] Small code fix for differential_privacy.py --- ethics/differential_privacy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ethics/differential_privacy.py b/ethics/differential_privacy.py index 0a1a6ab..97f4c7f 100644 --- 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