Skip to content

Commit 8e8304e

Browse files
committed
Fix failing tests, return values from an integer tensor are integers, not strings
1 parent b43ecc4 commit 8e8304e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/basic_tests.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def test_set_tensor(env):
5555
con.execute_command('AI.TENSORSET', 'x', 'INT32', 2, 'VALUES', 2, 3)
5656
tensor = con.execute_command('AI.TENSORGET', 'x', 'VALUES')
5757
values = tensor[-1]
58-
env.assertEqual(values, [b'2', b'3'])
58+
env.assertEqual(values, [2, 3])
5959

6060
def test_set_tensor_multiproc(env):
6161
run_test_multiproc(env, 10,

0 commit comments

Comments
 (0)