🐛 Describe the bug

After my PR is landed, the nightly releases become broken. See https://github.com/pytorch/data/actions/runs/3281029405

• Fix the test
• Investigate why the test is failing but succeeded on PR. And, the integration test is unrelated to my PR

Versions

main branch