update pytorch unit test

Deng · Deng · commit 2581ae76f45d · 2020-03-04T15:20:43.000-08:00
diff --git a/tests/unit/test_pytorch.py b/tests/unit/test_pytorch.py
@@ -348,7 +348,7 @@ def test_model_image_accelerator(sagemaker_session):
     model = PyTorchModel(
         MODEL_DATA, role=ROLE, entry_point=SCRIPT_PATH, sagemaker_session=sagemaker_session
     )
-    predictor = model.deploy(1, CPU)
+    predictor = model.deploy(1, CPU, accelerator_type=ACCELERATOR_TYPE)
     assert isinstance(predictor, PyTorchPredictor)
 
 

Original file line number	Diff line number	Diff line change
`@@ -348,7 +348,7 @@ def test_model_image_accelerator(sagemaker_session):`
`348`	`348`	`model = PyTorchModel(`
`349`	`349`	`MODEL_DATA, role=ROLE, entry_point=SCRIPT_PATH, sagemaker_session=sagemaker_session`
`350`	`350`	`)`
`351`		`- predictor = model.deploy(1, CPU)`
	`351`	`+ predictor = model.deploy(1, CPU, accelerator_type=ACCELERATOR_TYPE)`
`352`	`352`	`assert isinstance(predictor, PyTorchPredictor)`
`353`	`353`
`354`	`354`