udacity#7 Python 3.* should use 'from unittest import mock

', Python 2.* use: 'from mock import mock '
caofangkun · Jan 19, 2018 · bf5d16f · bf5d16f
1 parent 0591ddd
commit bf5d16f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/face-generation/problem_unittests.py b/face-generation/problem_unittests.py
@@ -1,5 +1,8 @@
 from copy import deepcopy
-from unittest import mock
+try:
+    from unittest import mock
+except ImportError:
+    from mock import mock
 import tensorflow as tf