Skip to content

Commit ffbcf53

Browse files
committed
Update estimator env check when updating model env
1 parent 597fe1d commit ffbcf53

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/stepfunctions/steps/sagemaker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ def get_expected_model(self, model_name=None):
157157
model.name = model_name
158158
else:
159159
model.name = self.job_name
160-
if self.estimator.environment is not None:
160+
if self.estimator.environment:
161161
model.env = self.estimator.environment
162162
model.model_data = self.output()["ModelArtifacts"]["S3ModelArtifacts"]
163163
return model

0 commit comments

Comments
 (0)