Commit ffbcf53

committed

Update estimator env check when updating model env

1 parent 597fe1d commit ffbcf53Copy full SHA for ffbcf53

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -157,7 +157,7 @@ def get_expected_model(self, model_name=None):`
`157`	`157`	`model.name = model_name`
`158`	`158`	`else:`
`159`	`159`	`model.name = self.job_name`
`160`		`- if self.estimator.environmentisnotNone:`
	`160`	`+ if self.estimator.environment:`
`161`	`161`	`model.env = self.estimator.environment`
`162`	`162`	`model.model_data = self.output()["ModelArtifacts"]["S3ModelArtifacts"]`
`163`	`163`	`return model`

Comments

(0)