Fix travis

hunkim · hunkim · commit 4a521c5b7111 · 2017-03-09T13:00:18.000+09:00
diff --git a/.travis.yml b/.travis.yml
@@ -13,6 +13,7 @@ script:
   - sed -i -- 's/range(total_batch)/range(1)/g' *lab-*.py # change range to 1 for quick testing
   - sed -i -- 's/plt.show/#plt.show/g' *lab-*.py # no plt
   - sed -i -- 's/plt.plot/#plt.plot/g' *lab-*.py # no plt
+  - sed -i -- 's/plt.imshow/#plt.imshow/g' *lab-*.py # no plt
   - sed -i -- 's/nb_epoch = 12/nb_epoch = 1/g' *lab-*.py # change range to 1 for quick testing
   - sed -i -- 's/plot(/# plot (/g' *lab-*.py # change range to 1 for quick testing
   - sed -i -- 's/from keras.utils.visualize_util import plot/#/g' *lab-*.py
diff --git a/lab-07-2-learning_rate_and_evaluation.py b/lab-07-2-learning_rate_and_evaluation.py
@@ -54,8 +54,7 @@
     print("Prediction: ", sess.run(
         tf.argmax(hypothesis, 1), feed_dict={X: mnist.test.images[r:r + 1]}))
 
-    plt.imshow(mnist.test.images[r:r + 1].
-               reshape(28, 28), cmap='Greys', interpolation='nearest')
+    plt.imshow(mnist.test.images[r:r + 1].reshape(28, 28), cmap='Greys', interpolation='nearest')
     plt.show()
 
 
