Fix autogluon license test (#4977)

* Fix license test for Autogluon images

* Python formatting fix

* Resolve comments

* Add Pytorch as framework

Author: Atharva-Rajan-Kale

test/dlc_tests/sanity/test_pre_release.py

@@ -1078,6 +1078,9 @@ def test_license_file(image):
     framework, version = get_framework_and_version_from_tag(image)
     if framework == "autogluon":
         short_version = get_pytorch_version_from_autogluon_image(image)
+        # Default to pytorch framework for autogluon since autogluon is built on top of pytorch
+        # and uses the same license file structure in S3
+        framework = "pytorch"
     else:
         short_version = re.search(r"(\d+\.\d+)", version).group(0)
     LICENSE_FILE_BUCKET = "aws-dlc-licenses"