Skip to content

Commit 1d79ca6

Browse files
imujjwal96pengk19
authored andcommitted
fix: skip local file check for TF requirements file when source_dir is an S3 URI (aws#798)
1 parent ad8fdf4 commit 1d79ca6

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/tensorflow/estimator.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,9 @@ def _validate_requirements_file(self, requirements_file):
283283
if not self.source_dir:
284284
raise ValueError('Must specify source_dir along with a requirements file.')
285285

286+
if self.source_dir.lower().startswith('s3://'):
287+
return
288+
286289
if os.path.isabs(requirements_file):
287290
raise ValueError('Requirements file {} is not a path relative to source_dir.'.format(
288291
requirements_file))

0 commit comments

Comments
 (0)