Skip to content

Commit 32b9c79

Browse files
fix format issue
1 parent df4bbdc commit 32b9c79

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/sagemaker/huggingface/training_compiler/config.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,8 +107,9 @@ def validate(cls, estimator):
107107
Version(estimator.pytorch_version) in SpecifierSet("> 1.11")
108108
):
109109
error_helper_string = (
110-
"SageMaker Training Compiler is only supported with HuggingFace PyTorch 1.9-1.11 "
111-
" Received pytorch_version={} which is unsupported."
110+
"SageMaker Training Compiler is only supported "
111+
"with HuggingFace PyTorch 1.9-1.11. "
112+
"Received pytorch_version={} which is unsupported."
112113
)
113114
raise ValueError(error_helper_string.format(estimator.pytorch_version))
114115

src/sagemaker/tensorflow/training_compiler/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def validate(cls, estimator):
9292
super(TrainingCompilerConfig, cls).validate(estimator)
9393

9494
if estimator.framework_version:
95-
if not Version(estimator.framework_version) in SpecifierSet(
95+
if Version(estimator.framework_version) not in SpecifierSet(
9696
f">= {cls.MIN_SUPPORTED_VERSION}", f"<= {cls.MAX_SUPPORTED_VERSION}"
9797
):
9898
error_helper_string = (

0 commit comments

Comments
 (0)