Fix jit tests assuming sample_inputs is a list (#69975)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69975
cc mruberry
Test Plan: Imported from OSS
Reviewed By: ejguan
Differential Revision: D33172952
Pulled By: mruberry
fbshipit-source-id: 1f8bb49179f7fbd0fec5e7344e8c213484518e27