Skip to content

Commit 6edc62d

Browse files
author
Che Vu Gia Hy
committed
TF: Fix validation of default framework_version
Which should be passed modified variable.
1 parent 8a6b784 commit 6edc62d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/tensorflow/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -302,7 +302,7 @@ def __init__(
302302
self._validate_args(
303303
py_version=py_version,
304304
script_mode=script_mode,
305-
framework_version=framework_version,
305+
framework_version=self.framework_version,
306306
training_steps=training_steps,
307307
evaluation_steps=evaluation_steps,
308308
requirements_file=requirements_file,

0 commit comments

Comments
 (0)