Skip to content

Commit f636a7b

Browse files
committed
[fix] fixed gpu ci.
1 parent b24ad60 commit f636a7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/tests_onnx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def test_onnx_modelrun_mnist_autobatch(env):
175175
ret = con.execute_command('AI.MODELGET', 'm', 'META')
176176
env.assertEqual(len(ret), 14)
177177
# TODO: enable me. CI is having issues on GPU asserts of ONNX and CPU
178-
if DEVICE == "CPU":
178+
if DEVICE == "CPU":
179179
env.assertEqual(ret[1], b'ONNX')
180180
env.assertEqual(ret[3], b'CPU')
181181
env.assertEqual(ret[5], b'')

0 commit comments

Comments
 (0)