Skip to content

Commit d1e2ab7

Browse files
imujjwal96laurenyu
authored andcommitted
fix: skip local file check for TF requirements file when source_dir is an S3 URI (#798)
1 parent 24a20b3 commit d1e2ab7

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/tensorflow/estimator.py

+3
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)