From b6c8ecbc7d5aa4ad0dab153188c6ad1283022ee2 Mon Sep 17 00:00:00 2001 From: Jakob Jordan Date: Tue, 7 Jul 2020 15:01:55 +0200 Subject: [PATCH] Actually use `to_torch` in `_test_to_torch` --- test/test_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_node.py b/test/test_node.py index 6fccba4a..9a1a5a90 100644 --- 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):