From 968f933c107b5cb6c5d7c8983796d2d6aa17fe0e Mon Sep 17 00:00:00 2001 From: Xinwei Date: Wed, 15 Jan 2025 13:14:49 -0800 Subject: [PATCH] fix import issue --- research/disk_optimizer/KFprivacy_engine.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/research/disk_optimizer/KFprivacy_engine.py b/research/disk_optimizer/KFprivacy_engine.py index ac0626ee..c890c224 100644 --- a/research/disk_optimizer/KFprivacy_engine.py +++ b/research/disk_optimizer/KFprivacy_engine.py @@ -4,7 +4,7 @@ from opacus.privacy_engine import PrivacyEngine from torch import optim -from optimizers import KF_DPOptimizer, get_optimizer_class +from .optimizers import KF_DPOptimizer, get_optimizer_class class KF_PrivacyEngine(PrivacyEngine):