File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments