From 30c319f2a0bce83c65b8f976b34d41d44c857a50 Mon Sep 17 00:00:00 2001
From: Vinh Khuc <knvinh@gmail.com>
Date: Sat, 1 Jul 2017 20:47:04 -0700
Subject: [PATCH] Fixed missing pytorch.

---
 .travis.yml | 29 ++++++++++++++++++++---------
 README.md   |  2 +-
 2 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 76640d9..8bebb94 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,12 +1,23 @@
+os:
+  - linux
 language: python
 python:
-  - "2.7"
-  - "3.6"
+  - 2.7
+  - 3.6
+cache: bundler
+install:
+  - if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]]; then
+      pip install -q http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp27-none-linux_x86_64.whl;
+    else
+      pip install -q http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp36-cp36m-linux_x86_64.whl;
+    fi
+  - pip install -q torchvision
+  - pip install -q numpy==1.13.0
 script:
-  - python 0_multiply.py
-  - 1_linear_regression.py
-  - 2_logistic_regression.py
-  - 3_neural_net.py
-  - 4_modern_neural_net.py
-  - 5_convolutional_net.py
-  - 6_lstm.py
+  - timeout 5 python 0_multiply.py
+  - timeout 5 python 1_linear_regression.py
+  - timeout 5 python 2_logistic_regression.py
+  - timeout 5 python 3_neural_net.py
+  - timeout 5 python 4_modern_neural_net.py
+  - timeout 5 python 5_convolutional_net.py
+  - timeout 5 python 6_lstm.py
diff --git a/README.md b/README.md
index f5b4caf..47a0127 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-[![Build Status](https://travis-ci.org/vinhkhuc/PyTorch-Mini-Tutorials.svg?branch=master)](https://travis-ci.org/vinhkhuc/PyTorch-Mini-Tutorials)
+<!-- [![Build Status](https://travis-ci.org/vinhkhuc/PyTorch-Mini-Tutorials.svg?branch=master)](https://travis-ci.org/vinhkhuc/PyTorch-Mini-Tutorials) -->
 
 Minimal tutorials for [PyTorch](https://github.com/pytorch/pytorch) adapted
 from Alec Radford's [Theano tutorials](https://github.com/Newmu/Theano-Tutorials).