conf

pchavanne · Dec 10, 2016 · 5b54e12 · 5b54e12
1 parent 0abdb8c
commit 5b54e12
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 9 deletions.
diff --git a/tests/test_layers.py b/tests/test_layers.py
@@ -77,7 +77,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (16, 3, None, None, 10)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     def test_reshape(self, reshape_layer, input_layer):
         layer = reshape_layer(input_layer, (16, 3, 5, 7, 2, 5))
@@ -104,7 +104,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (2, 3, 4, 5,)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     def test_output_shape(self, flatten_layer, input_layer):
         layer = flatten_layer(input_layer)
@@ -132,7 +132,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (10, 20)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     @pytest.fixture
     def layer(self, dense_layer, input_layer):
@@ -178,7 +178,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (10, 20)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     @pytest.fixture
     def hp(self):
@@ -237,7 +237,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (None, 20)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     @pytest.fixture
     def layer(self, dropout, input_layer):
@@ -271,7 +271,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (10, 20)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     @pytest.fixture
     def layer(self, dropconnect, input_layer):
@@ -304,7 +304,7 @@ def input_data(self):
     def input_layer(self, input_data):
         from yadll.layers import InputLayer
         shape = (10, 20)
-        return InputLayer(shape, input_var=input_data)
+        return InputLayer(shape, input=input_data)
 
     @pytest.fixture
     def layer(self, pool_layer, input_layer):

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -54,7 +54,7 @@ def model_y_2D(self, data_y_2D, hp):
     @pytest.fixture(scope='module')
     def input(self):
         from yadll.layers import InputLayer
-        return InputLayer(shape=(None, 25))
+        return InputLayer(input_shape=(None, 25))
 
     @pytest.fixture(scope='module')
     def layer(self, input):

diff --git a/tests/test_network.py b/tests/test_network.py
@@ -20,7 +20,7 @@ def x(self):
     @pytest.fixture
     def input(self, x):
         from yadll.layers import InputLayer
-        return InputLayer(shape=(None, 25), input_var=x)
+        return InputLayer(input_shape=(None, 25), input=x)
 
     @pytest.fixture
     def layer(self, input):