fix import issue

pytorch · Jan 15, 2025 · 968f933 · 968f933
1 parent 5251d29
commit 968f933
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):