Skip to content

Commit ae85184

Browse files
committed
fix: remove debug print and add JsonGet input type check
1 parent 0a1ac2f commit ae85184

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

src/sagemaker/workflow/functions.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -70,6 +70,10 @@ def expr(self):
7070
name = self.property_file.name
7171
else:
7272
name = self.property_file
73+
74+
if not isinstance(self.processing_step_name, str):
75+
raise ValueError("processing_step_name passed in is not instance of a str")
76+
7377
return {
7478
"Std:JsonGet": {
7579
"PropertyFile": {"Get": f"Steps.{self.processing_step_name}.PropertyFiles.{name}"},

src/sagemaker/workflow/pipeline.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,6 @@ def definition(self) -> str:
242242
request_dict["PipelineExperimentConfig"] = interpolate(
243243
request_dict["PipelineExperimentConfig"]
244244
)
245-
print(request_dict["Steps"])
246245
request_dict["Steps"] = interpolate(request_dict["Steps"])
247246

248247
return json.dumps(request_dict)

0 commit comments

Comments
 (0)