File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed
huggingface/training_compiler
tensorflow/training_compiler Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -107,8 +107,9 @@ def validate(cls, estimator):
107
107
Version (estimator .pytorch_version ) in SpecifierSet ("> 1.11" )
108
108
):
109
109
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."
112
113
)
113
114
raise ValueError (error_helper_string .format (estimator .pytorch_version ))
114
115
Original file line number Diff line number Diff line change @@ -92,7 +92,7 @@ def validate(cls, estimator):
92
92
super (TrainingCompilerConfig , cls ).validate (estimator )
93
93
94
94
if estimator .framework_version :
95
- if not Version (estimator .framework_version ) in SpecifierSet (
95
+ if Version (estimator .framework_version ) not in SpecifierSet (
96
96
f">= { cls .MIN_SUPPORTED_VERSION } " , f"<= { cls .MAX_SUPPORTED_VERSION } "
97
97
):
98
98
error_helper_string = (
You can’t perform that action at this time.
0 commit comments