Add specific name

Author: saimidu

test/integration/__init__.py

@@ -75,6 +75,7 @@
     default_traced_resnet_script,
     os.path.join(default_traced_resnet_dir, "traced_resnet18.pt"),
     default_traced_resnet_dir,
+    filename="traced_resnet18.tar.gz",
     script_path="code",
 )
 

test/integration/sagemaker/test_default_inference.py

@@ -75,7 +75,11 @@ def test_default_inference_any_model_name_gpu(sagemaker_session, image_uri, inst
     # Scripted model is serialized with torch.jit.save().
     # Default inference test doesn't need to instantiate model definition
     _test_default_inference(
-        sagemaker_session, image_uri, instance_type, default_model_traced_resnet18_tar, default_traced_resnet_script
+        sagemaker_session,
+        image_uri,
+        instance_type,
+        default_model_traced_resnet18_tar,
+        default_traced_resnet_script,
     )