Whoops, fix that bug in test

brendanhasz · Dec 23, 2020 · 0ffa3fc · 0ffa3fc
1 parent 467803f
commit 0ffa3fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/tensorflow/utils/test_casting.py b/tests/unit/tensorflow/utils/test_casting.py
@@ -15,7 +15,7 @@ def test_to_numpy():
     assert o.shape[1] == 2
 
     x = tf.random.normal([5, 2])
-    o = casting.to_tensor(x)
+    o = casting.to_numpy(x)
     assert isinstance(o, np.ndarray)
     assert o.ndim == 2
     assert o.shape[0] == 5