Correct docs to reflect that upload_data would create only default bucket. (aws#373)

Author: nadiaya

src/sagemaker/session.py

@@ -119,8 +119,8 @@ def upload_data(self, path, bucket=None, key_prefix='data'):
         Args:
             path (str): Path (absolute or relative) of local file or directory to upload.
             bucket (str): Name of the S3 Bucket to upload to (default: None). If not specified, the
-                default bucket of the ``Session`` is used. If the bucket does not exist, the ``Session``
-                creates the bucket.
+                default bucket of the ``Session`` is used (if default bucket does not exist, the ``Session``
+                creates it).
             key_prefix (str): Optional S3 object key name prefix (default: 'data'). S3 uses the prefix to
                 create a directory structure for the bucket content that it display in the S3 console.
 

tests/unit/test_pytorch.py

@@ -24,14 +24,15 @@
 from sagemaker.pytorch import PyTorch
 from sagemaker.pytorch import PyTorchPredictor, PyTorchModel
 
-from tests.integ import DATA_DIR, PYTHON_VERSION
 
+DATA_DIR = os.path.join(os.path.dirname(__file__), '..', 'data')
 SCRIPT_PATH = os.path.join(DATA_DIR, 'dummy_script.py')
 TIMESTAMP = '2017-11-06-14:14:15.672'
 TIME = 1507167947
 BUCKET_NAME = 'mybucket'
 INSTANCE_COUNT = 1
 INSTANCE_TYPE = 'ml.c4.4xlarge'
+PYTHON_VERSION = 'py' + str(sys.version_info.major)
 IMAGE_NAME = 'sagemaker-pytorch'
 JOB_NAME = '{}-{}'.format(IMAGE_NAME, TIMESTAMP)
 IMAGE_URI_FORMAT_STRING = "520713654638.dkr.ecr.{}.amazonaws.com/{}:{}-{}-{}"