Skip to content

Commit 320d05b

Browse files
committed
fix: logic error in trcomp initialization in HF estimator
1 parent cbac484 commit 320d05b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sagemaker/huggingface/estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,6 +190,8 @@ def __init__(
190190
entry_point, source_dir, hyperparameters, image_uri=image_uri, **kwargs
191191
)
192192

193+
self.distribution = distribution or {}
194+
193195
if compiler_config is not None:
194196
if not isinstance(compiler_config, TrainingCompilerConfig):
195197
error_string = (
@@ -200,8 +202,6 @@ def __init__(
200202
raise ValueError(error_string)
201203
if compiler_config:
202204
compiler_config.validate(self)
203-
204-
self.distribution = distribution or {}
205205
self.compiler_config = compiler_config
206206

207207
def _validate_args(self, image_uri):

0 commit comments

Comments
 (0)