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/3075#018dd308-0f21-46e4-a191-84d23d0c907c
• https://buildkite.com/ray-project/postmerge/builds/3026#018dcb54-71b6-48b9-b2b0-5778438554f2
• https://buildkite.com/ray-project/postmerge/builds/3010#018dc777-03f7-4d11-af20-a43d7d515f3b
• https://buildkite.com/ray-project/postmerge/builds/3005#018dc62e-029b-4fed-ab44-db068d8ccc8a
• https://buildkite.com/ray-project/postmerge/builds/3003#018dc4ce-efa6-4765-be23-334525ae53bd

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/3097#018dd78f-cf80-4799-9d26-611239a67400