From 0b933c1699e208a4c58df310a6eaddd798a5383b Mon Sep 17 00:00:00 2001
From: jin-xiulang <jinxiulang@huawei.com>
Date: Fri, 24 Jul 2020 15:37:57 +0800
Subject: [PATCH] Fix a bug.

---
 example/mnist_demo/lenet5_dp_ada_gaussian.py | 2 +-
 mindarmour/fuzzing/fuzzing.py                | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/example/mnist_demo/lenet5_dp_ada_gaussian.py b/example/mnist_demo/lenet5_dp_ada_gaussian.py
index 23ac851..484c733 100644
--- a/example/mnist_demo/lenet5_dp_ada_gaussian.py
+++ b/example/mnist_demo/lenet5_dp_ada_gaussian.py
@@ -34,7 +34,7 @@ from mindarmour.diff_privacy import PrivacyMonitorFactory
 from mindarmour.diff_privacy import NoiseMechanismsFactory
 from mindarmour.utils.logger import LogUtil
 from lenet5_net import LeNet5
-from lenet5_config import mnist_cfg as cfg
+from dp_ada_gaussian_config import mnist_cfg as cfg
 
 LOGGER = LogUtil.get_instance()
 LOGGER.set_level('INFO')
diff --git a/mindarmour/fuzzing/fuzzing.py b/mindarmour/fuzzing/fuzzing.py
index 10a119f..e613c63 100644
--- a/mindarmour/fuzzing/fuzzing.py
+++ b/mindarmour/fuzzing/fuzzing.py
@@ -95,7 +95,7 @@ class Fuzzing:
                     mutate_tests.append(mutate_test)
             if not mutate_tests:
                 mutate_tests.append(seed)
-            return np.array(mutate_tests)
+        return np.array(mutate_tests)
 
     def fuzzing(self, coverage_metric='KMNC'):
         """