CI test linux://rllib:examples/action_masking_torch is flaky

[can-anyscale]

CI test linux://rllib:examples/action_masking_torch is flaky. Recent failures:

• https://buildkite.com/ray-project/postmerge/builds/2964#018da9b8-78bd-4d5c-8ffc-518cf39d0459
• https://buildkite.com/ray-project/postmerge/builds/2937#018da334-65e7-4fe1-a4de-b7b9e0ef34ac
• https://buildkite.com/ray-project/postmerge/builds/2928#018d9f6d-f76c-4483-9517-02d73fc1e278
• https://buildkite.com/ray-project/postmerge/builds/2915#018d9378-7fd7-40b4-afdf-6e04144acd52
• https://buildkite.com/ray-project/postmerge/builds/2894#018d86c3-20fe-4ba4-8166-edc396f23abe

DataCaseName-linux://rllib:examples/action_masking_torch-END Managed by OSS Test Policy

[can-anyscale]

Test passed on latest run: https://buildkite.com/ray-project/postmerge/builds/3049#018dce3a-25e8-457d-a0fd-6630f15b9181