Update test_average_precision.py to resolve issue #3335

[puhuk]

Related to #3335

Description: Resolved the issue with test_average_precision.py when running in distributed mode, as reported in #3335.

Check list:

• New tests are added (if a new feature is added)
• New doc strings: description and/or example code are in RST format
• Documentation is updated (if required)

[vfdev-5]

Thanks for the PR @puhuk
I left few comments how to make it better

[vfdev-5]

Let's keep track of https://github.com/pytorch/ignite/actions/runs/13568080551/job/37925851877?pr=3337 job and see whether there are related failing tests. If some related tests are failing we need to provide a fix in this PR.

[puhuk]

Let's keep track of https://github.com/pytorch/ignite/actions/runs/13568080551/job/37925851877?pr=3337 job and see whether there are related failing tests. If some related tests are failing we need to provide a fix in this PR.

Got it! Keeping an eye on it! If anything breaks, I’ll be ready to fix it. 🚀

[puhuk]

@vfdev-5
I have a question regarding the mps-tests. It reports an error in test_precision.
Why does this error occur even though I only modified test_average_precision?
Are you suggesting that test_precision should also be fixed in this PR?

[vfdev-5]

@vfdev-5 I have a question regarding the mps-tests. It reports an error in test_precision. Why does this error occur even though I only modified test_average_precision? Are you suggesting that test_precision should also be fixed in this PR?

This should be done in #3334
which is almost finished (remains one failure in one test). Once that PR is merged you can rebase your PR on the master and the unrelated failures should disappear.

[vfdev-5]

Thanks @puhuk ! LGTM