Skip to content

Commit 2083bac

Browse files
fix format issue
1 parent 668064b commit 2083bac

File tree

1 file changed

+3
-1
lines changed
  • src/sagemaker/huggingface/training_compiler

1 file changed

+3
-1
lines changed

src/sagemaker/huggingface/training_compiler/config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,9 @@ def validate(cls, estimator):
103103
super(TrainingCompilerConfig, cls).validate(estimator)
104104

105105
if estimator.pytorch_version:
106-
if (Version(estimator.pytorch_version) in SpecifierSet("< 1.9")) or (Version(estimator.pytorch_version) in SpecifierSet("> 1.11")):
106+
if (Version(estimator.pytorch_version) in SpecifierSet("< 1.9")) or (
107+
Version(estimator.pytorch_version) in SpecifierSet("> 1.11")
108+
):
107109
error_helper_string = (
108110
"Training Compiler is only valid between HuggingFace PyTorch 1.9-1.11 "
109111
"for SageMaker Training Compiler."

0 commit comments

Comments
 (0)