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/2868#018d80bd-cf08-468d-90e3-cf9513be47ad
• https://buildkite.com/ray-project/postmerge/builds/2825#018d6a8e-b3ad-46f5-8133-05fec51a8e24
• https://buildkite.com/ray-project/postmerge/builds/2808#018d625b-f1c3-4658-b098-7c92ff8be744
• https://buildkite.com/ray-project/postmerge/builds/2801#018d61e0-c800-491b-a680-d457aa537aae
• https://buildkite.com/ray-project/postmerge/builds/2787#018d6041-ce79-4d24-8385-5e7b9ccd371d

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/2960#018da748-2398-4737-8cfb-0df9908c9740