Skip to content

Commit 5657fd4

Browse files
committed
revert previous change
1 parent 223d4de commit 5657fd4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -622,7 +622,7 @@ def _compilation_job_config(
622622
if (
623623
framework.lower() == "pytorch"
624624
or framework.lower() == "tensorflow"
625-
and re.match("(?=^ml_)(?!ml_inf)", str(target_instance_type)) is not None
625+
and re.match("(?=^ml_)(?!ml_inf)", target_instance_type) is not None
626626
and framework_version is not None
627627
):
628628
input_model_config["FrameworkVersion"] = utils.get_short_version(framework_version)

0 commit comments

Comments
 (0)