Do not change default version, add latest version instead.

Author: nadiaya

src/sagemaker/pytorch/defaults.py

@@ -12,5 +12,5 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-PYTORCH_VERSION = '1.0'
+PYTORCH_VERSION = '0.4'
 PYTHON_VERSION = 'py3'

src/sagemaker/pytorch/estimator.py

@@ -29,6 +29,8 @@ class PyTorch(Framework):
 
     __framework_name__ = "pytorch"
 
+    LATEST_VERSION = '1.0'
+
     def __init__(self, entry_point, source_dir=None, hyperparameters=None, py_version=PYTHON_VERSION,
                  framework_version=None, image_name=None, **kwargs):
         """

tests/conftest.py

@@ -23,7 +23,7 @@
 from sagemaker.chainer import Chainer
 from sagemaker.local import LocalSession
 from sagemaker.mxnet import MXNet
-from sagemaker.pytorch.defaults import PYTORCH_VERSION
+from sagemaker.pytorch import PyTorch
 from sagemaker.rl import RLEstimator
 from sagemaker.sklearn.defaults import SKLEARN_VERSION
 from sagemaker.tensorflow.defaults import TF_VERSION
@@ -37,7 +37,7 @@ def pytest_addoption(parser):
     parser.addoption('--boto-config', action='store', default=None)
     parser.addoption('--chainer-full-version', action='store', default=Chainer.LATEST_VERSION)
     parser.addoption('--mxnet-full-version', action='store', default=MXNet.LATEST_VERSION)
-    parser.addoption('--pytorch-full-version', action='store', default=PYTORCH_VERSION)
+    parser.addoption('--pytorch-full-version', action='store', default=PyTorch.LATEST_VERSION)
     parser.addoption('--rl-coach-full-version', action='store',
                      default=RLEstimator.COACH_LATEST_VERSION)
     parser.addoption('--rl-ray-full-version', action='store',