Fixed issue with jitted AA transforms in v2 and added tests

[vfdev-5]

Error reported in #7159

2023-08-15T10:23:55.6431105Z E   RuntimeError: The following operation failed in the TorchScript interpreter.
2023-08-15T10:23:55.6431482Z E   Traceback of TorchScript (most recent call last):
2023-08-15T10:23:55.6431884Z E     File "/pytorch/vision/torchvision/transforms/autoaugment.py", line 268, in forward
2023-08-15T10:23:55.6432243Z E                   fill = [float(fill)] * channels
2023-08-15T10:23:55.6432525Z E               elif fill is not None:
2023-08-15T10:23:55.6432816Z E                   fill = [float(f) for f in fill]
2023-08-15T10:23:55.6433084Z E                           ~~~~~ <--- HERE
2023-08-15T10:23:55.6433303Z E       
2023-08-15T10:23:55.6433631Z E           transform_id, probs, signs = self.get_params(len(self.policies))
2023-08-15T10:23:55.6433925Z E   RuntimeError: stod

• https://hud.pytorch.org/pytorch/vision/pull/7159?sha=699529aae13ab6739d336135fb08e1dd9acccfbb

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/vision/7839

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

✅ 1 Unrelated Failure

As of commit 0fe1c54 with merge base 37081ee ():

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

• pytorch/vision / wheel-py3_8-cpu-aarch64 (gh)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[pmeier]

LGTM if CI is green. Thanks Victor!

[NicolasHug]

@vfdev-5 @pmeier instead of adding new tests here, shouldn't we be writing the new ones in test_transforms_v2_refactored.py?

We're going to delete this file, right?

[vfdev-5]

@NicolasHug can we merge this PR and address your comment later ?

[pmeier]

We're going to delete this file, right?

Yes. Indeed, all new tests should go into test_transforms_v2_refactored.py. No strong opinion of doing this now or later.

[NicolasHug]

OK, let's do it later

[github-actions]

Hey @vfdev-5!

You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py