Skip to content

Commit c5d37a7

Browse files
committed
fix unit test
1 parent fc70d20 commit c5d37a7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/workflow/steps.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,10 @@ class CacheConfig:
119119
@property
120120
def config(self):
121121
"""Configures caching in pipeline steps."""
122-
config = {'Enabled': self.enable_caching}
122+
config = {"Enabled": self.enable_caching}
123123
if self.expire_after is not None:
124-
config['ExpireAfter'] = self.expire_after
125-
return {'CacheConfig': config}
124+
config["ExpireAfter"] = self.expire_after
125+
return {"CacheConfig": config}
126126

127127

128128
class TrainingStep(Step):

0 commit comments

Comments
 (0)