From 8b6cb52bbdb27ab73ed1181c61b3b892824942df Mon Sep 17 00:00:00 2001 From: Ujjwal Bhardwaj Date: Tue, 14 May 2019 17:41:55 +0000 Subject: [PATCH] fix: TF estimator s3 requirements file validation --- src/sagemaker/tensorflow/estimator.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/sagemaker/tensorflow/estimator.py b/src/sagemaker/tensorflow/estimator.py index 67e7745b07..983e743315 100644 --- a/src/sagemaker/tensorflow/estimator.py +++ b/src/sagemaker/tensorflow/estimator.py @@ -283,6 +283,9 @@ def _validate_requirements_file(self, requirements_file): if not self.source_dir: raise ValueError('Must specify source_dir along with a requirements file.') + if self.source_dir.lower().startswith('s3://'): + return + if os.path.isabs(requirements_file): raise ValueError('Requirements file {} is not a path relative to source_dir.'.format( requirements_file))