Fix tests

malmaud · Jun 27, 2017 · 8beba47 · 8beba47
1 parent 6980b1c
commit 8beba47
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/test/core.jl b/test/core.jl
@@ -77,7 +77,9 @@ end
             unused = get_variable("unused", [], Float64)
             used = get_variable("used", [], Float64)
             loss = used.^2
-            optimizer = train.minimize(train.AdamOptimizer(), loss)
+            grad = gradients(loss, unused)
+            @test grad === nothing
+            # optimizer = train.minimize(ætrain.AdamOptimizer(), loss)
             # This would have thrown an error if Disconnected gradients were causing issues
         end
     end

diff --git a/test/meta.jl b/test/meta.jl
@@ -34,7 +34,7 @@ end
                 fq = TensorFlow.FIFOQueue(10, [Int32, Int64]); #This datatype should be an Op
                 cc = nn.rnn_cell.GRUCell(40) #this Datatype should not be an Op
 
-                X = get_variable([50], Float64)
+                X = get_variable([1, 50], Float64)
                 variable_scope("logisitic_model") do
                      W = get_variable([50, 10], Float64)
                      B = get_variable([10], Float64)