[TEST] Comptiable with python3.5 (#3675)

0365e50a · lixiaoquan · Tianqi Chen · 11da1ca3 · 0365e50a
Commit 0365e50a authored Jul 31, 2019 by lixiaoquan Committed by Tianqi Chen Jul 31, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

tests/python/frontend/tensorflow/test_forward.py
+2 -2

No files found.
--- a/tests/python/frontend/tensorflow/test_forward.py
+++ b/tests/python/frontend/tensorflow/test_forward.py
@@ -922,9 +922,9 @@ def _test_unstack(ip_shape, axis, dtype):

    tf.reset_default_graph()
    in_data = tf.placeholder(dtype, ip_shape, name="in_data")
-    tf.unstack(in_data, axis=axis)
+    unstack = tf.unstack(in_data, axis=axis)

-    compare_tf_with_tvm([np_data], ['in_data:0'], [f'unstack:{n}' for n in range(ip_shape[axis])])
+    compare_tf_with_tvm([np_data], ['in_data:0'], [n.name for n in unstack])

    tf.reset_default_graph()
    in_data = tf.placeholder(dtype, ip_shape, name="in_data")