From 0c4da595bb29ab3ce0b550ffa2706cdd63ce3f96 Mon Sep 17 00:00:00 2001
From: liuluobin <liuluobin@huawei.com>
Date: Thu, 20 Aug 2020 16:28:16 +0800
Subject: [PATCH] append reference in __init__.py

---
 mindarmour/diff_privacy/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/mindarmour/diff_privacy/__init__.py b/mindarmour/diff_privacy/__init__.py
index a54ce80..6dbd1e2 100644
--- a/mindarmour/diff_privacy/__init__.py
+++ b/mindarmour/diff_privacy/__init__.py
@@ -11,6 +11,7 @@ from .monitor.monitor import RDPMonitor
 from .monitor.monitor import ZCDPMonitor
 from .optimizer.optimizer import DPOptimizerClassFactory
 from .train.model import DPModel
+from .evaluation.membership_inference import MembershipInference
 
 __all__ = ['NoiseGaussianRandom',
            'NoiseAdaGaussianRandom',
@@ -21,4 +22,5 @@ __all__ = ['NoiseGaussianRandom',
            'RDPMonitor',
            'ZCDPMonitor',
            'DPOptimizerClassFactory',
-           'DPModel']
+           'DPModel',
+           'MembershipInference']