Merge pull request #10 from ixaxaar/hidden_layers

Test travis build
ixaxaar · Oct 29, 2017 · f8fc2f5 · f8fc2f5
2 parents d1763af + 7631472
commit f8fc2f5
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 9 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,7 @@
 # Differentiable Neural Computer, for Pytorch
 
+[![Build Status](https://travis-ci.org/ixaxaar/pytorch-dnc.svg?branch=master)](https://travis-ci.org/ixaxaar/pytorch-dnc) [![PyPI version](https://badge.fury.io/py/dnc.svg)](https://badge.fury.io/py/dnc)
+
 This is an implementation of [Differentiable Neural Computers](http://people.idsia.ch/~rupesh/rnnsymposium2016/slides/graves.pdf), described in the paper [Hybrid computing using a neural network with dynamic external memory, Graves et al.](https://www.nature.com/articles/nature20101)
 
 ## Install

diff --git a/test/test_gru.py b/test/test_gru.py
@@ -15,10 +15,9 @@
 import os
 import math
 import time
-sys.path.append('./src/')
-sys.path.insert(0, os.path.join('..', '..'))
+sys.path.insert(0, '.')
 
-from dnc.dnc import DNC
+from dnc import DNC
 from test_utils import generate_data, criterion
 
 

diff --git a/test/test_lstm.py b/test/test_lstm.py
@@ -15,10 +15,9 @@
 import os
 import math
 import time
-sys.path.append('./src/')
-sys.path.insert(0, os.path.join('..', '..'))
+sys.path.insert(0, '.')
 
-from dnc.dnc import DNC
+from dnc import DNC
 from test_utils import generate_data, criterion
 
 

diff --git a/test/test_rnn.py b/test/test_rnn.py
@@ -15,10 +15,9 @@
 import os
 import math
 import time
-sys.path.append('./src/')
-sys.path.insert(0, os.path.join('..', '..'))
+sys.path.insert(0, '.')
 
-from dnc.dnc import DNC
+from dnc import DNC
 from test_utils import generate_data, criterion