fix integ test

laurenyu · laurenyu · commit c5a19b898801 · 2020-07-01T09:03:03.000-07:00
diff --git a/tests/integ/test_inference_pipeline.py b/tests/integ/test_inference_pipeline.py
@@ -178,6 +178,7 @@ def test_inference_pipeline_model_deploy_and_update_endpoint(
         model = PipelineModel(
             models=[sparkml_model, xgb_model],
             role="SageMakerRole",
+            predictor_cls=Predictor,
             sagemaker_session=sagemaker_session,
         )
         predictor = model.deploy(1, alternative_cpu_instance_type, endpoint_name=endpoint_name)

Original file line number	Diff line number	Diff line change
`@@ -178,6 +178,7 @@ def test_inference_pipeline_model_deploy_and_update_endpoint(`
`178`	`178`	`model = PipelineModel(`
`179`	`179`	`models=[sparkml_model, xgb_model],`
`180`	`180`	`role="SageMakerRole",`
	`181`	`+ predictor_cls=Predictor,`
`181`	`182`	`sagemaker_session=sagemaker_session,`
`182`	`183`	`)`
`183`	`184`	`predictor = model.deploy(1, alternative_cpu_instance_type, endpoint_name=endpoint_name)`