qingpei's picture
Update test_infer.py
9a10e16 verified