Merge pull request #179 from jakobj/fix/test-to-torch

Actually use `to_torch` in `_test_to_torch`
Happy-Algorithms-League · Jul 7, 2020 · 66a8e5b · 66a8e5b
2 parents 6705227 + b6c8ecb
commit 66a8e5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_node.py b/test/test_node.py
@@ -65,7 +65,7 @@ def _test_to_numpy(genome, x, y_target):
 def _test_to_torch(genome, x, y_target):
     torch = pytest.importorskip("torch")
     graph = cgp.CartesianGraph(genome)
-    assert graph.to_numpy()(torch.Tensor(x).reshape(1, -1)) == pytest.approx(y_target)
+    assert graph.to_torch()(torch.Tensor(x).reshape(1, -1)) == pytest.approx(y_target)
 
 
 def _test_to_sympy(genome, x, y_target):