Fix class wrong import bug (microsoft#1271)

SparkSnail · Jul 8, 2019 · eea33e1 · eea33e1
1 parent f6d6667
commit eea33e1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/sdk/pynni/nni/networkmorphism_tuner/bayesian.py b/src/sdk/pynni/nni/networkmorphism_tuner/bayesian.py
@@ -29,8 +29,9 @@
 from scipy.optimize import linear_sum_assignment
 from sklearn.metrics.pairwise import rbf_kernel
 
+from nni.utils import OptimizeMode
 from nni.networkmorphism_tuner.graph_transformer import transform
-from nni.networkmorphism_tuner.utils import Constant, OptimizeMode
+from nni.networkmorphism_tuner.utils import Constant
 from nni.networkmorphism_tuner.layers import is_layer