add forgotten return statement

laurenyu · laurenyu · commit a63563130b63 · 2020-06-09T08:44:00.000-07:00
diff --git a/src/sagemaker/mxnet/estimator.py b/src/sagemaker/mxnet/estimator.py
@@ -239,6 +239,8 @@ def create_model(
                 self.entry_point if model._is_mms_version() else self._model_entry_point()
             )
 
+        return model
+
     @classmethod
     def _prepare_init_params_from_job_description(cls, job_details, model_channel_name=None):
         """Convert the job description to init params that can be handled by the

Original file line number	Diff line number	Diff line change
`@@ -239,6 +239,8 @@ def create_model(`
`239`	`239`	`self.entry_point if model._is_mms_version() else self._model_entry_point()`
`240`	`240`	`)`
`241`	`241`
	`242`	`+ return model`
	`243`	`+`
`242`	`244`	`@classmethod`
`243`	`245`	`def _prepare_init_params_from_job_description(cls, job_details, model_channel_name=None):`
`244`	`246`	`"""Convert the job description to init params that can be handled by the`