Skip to content

Commit e2c1918

Browse files
committed
Fix sanitizer tests
1 parent ec33683 commit e2c1918

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/tests_sanitizer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def test_sanitizer_dagrun_mobilenet_v1(env):
1212
if (not TEST_TF or not TEST_PT):
1313
return
1414
con = env.getConnection()
15-
mem_allocator = con.execute_command('info', 'memory')['mem_allocator']
15+
mem_allocator = con.info()['mem_allocator']
1616
if 'jemalloc' in mem_allocator:
1717
print("exiting sanitizer test given we're not using stdlib allocator")
1818
return
@@ -48,7 +48,7 @@ def test_sanitizer_modelrun_mobilenet_v1(env):
4848
if (not TEST_TF or not TEST_PT):
4949
return
5050
con = env.getConnection()
51-
mem_allocator = con.execute_command('info', 'memory')['mem_allocator']
51+
mem_allocator = con.info()['mem_allocator']
5252
if 'jemalloc' in mem_allocator:
5353
print("exiting sanitizer test given we're not using stdlib allocator")
5454
return

0 commit comments

Comments
 (0)