Use old pytorch image

samruds · samruds · commit c008915625b6 · 2024-03-19T17:41:03.000-07:00
diff --git a/tests/integ/sagemaker/serve/test_serve_pt_happy.py b/tests/integ/sagemaker/serve/test_serve_pt_happy.py
@@ -140,6 +140,7 @@ def model_builder_inference_spec_schema_builder(squeezenet_inference_spec, squee
         model_path=PYTORCH_SQUEEZENET_RESOURCE_DIR,
         inference_spec=squeezenet_inference_spec,
         schema_builder=squeezenet_schema,
+        image_uri="763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference:2.0.0-cpu-py310"
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -140,6 +140,7 @@ def model_builder_inference_spec_schema_builder(squeezenet_inference_spec, squee`
`140`	`140`	`model_path=PYTORCH_SQUEEZENET_RESOURCE_DIR,`
`141`	`141`	`inference_spec=squeezenet_inference_spec,`
`142`	`142`	`schema_builder=squeezenet_schema,`
	`143`	`+ image_uri="763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-inference:2.0.0-cpu-py310"`
`143`	`144`	`)`
`144`	`145`
`145`	`146`