Actually use to_torch in _test_to_torch

Happy-Algorithms-League · Jul 7, 2020 · b6c8ecb · b6c8ecb
1 parent 6705227
commit b6c8ecb
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):