Fixup test_trace_rules (#121351)
Summary:
Fixes
https://www.internalfb.com/intern/testinfra/diagnostics/7599824578133672.281475099376195.1709732674/
(for some reason this test didn't run in OSS)?
Reached out to Yanbo Liang for additional context:
{F1465435684}
Test Plan:
Local:
https://www.internalfb.com/intern/testinfra/testconsole/testrun/16325548673376150/
Differential Revision: D54605075
Pull Request resolved: https://github.com/pytorch/pytorch/pull/121351
Approved by: https://github.com/malfet, https://github.com/yanboliang