add py2 warning message

Author: Dan Choi

src/sagemaker/chainer/estimator.py

@@ -15,7 +15,8 @@
 import logging
 
 from sagemaker.estimator import Framework
-from sagemaker.fw_utils import framework_name_from_image, framework_version_from_tag, empty_framework_version_warning
+from sagemaker.fw_utils import framework_name_from_image, framework_version_from_tag, empty_framework_version_warning, \
+    python_deprecation_warning
 from sagemaker.chainer.defaults import CHAINER_VERSION
 from sagemaker.chainer.model import ChainerModel
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
@@ -92,7 +93,7 @@ def __init__(self, entry_point, use_mpi=None, num_processes=None, process_slots_
                                       image_name=image_name, **kwargs)
 
         if py_version == 'py2':
-            logger.warning('Chainer py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
         self.use_mpi = use_mpi

src/sagemaker/chainer/model.py

@@ -15,7 +15,7 @@
 import logging
 
 import sagemaker
-from sagemaker.fw_utils import create_image_uri, model_code_key_prefix
+from sagemaker.fw_utils import create_image_uri, model_code_key_prefix, python_deprecation_warning
 from sagemaker.model import FrameworkModel, MODEL_SERVER_WORKERS_PARAM_NAME
 from sagemaker.chainer.defaults import CHAINER_VERSION
 from sagemaker.predictor import RealTimePredictor, npy_serializer, numpy_deserializer
@@ -71,7 +71,7 @@ def __init__(self, model_data, role, entry_point, image=None, py_version='py3',
         super(ChainerModel, self).__init__(model_data, image, role, entry_point, predictor_cls=predictor_cls,
                                            **kwargs)
         if py_version == 'py2':
-            logger.warning('chainer py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
         self.framework_version = framework_version

src/sagemaker/fw_utils.py

@@ -36,6 +36,10 @@
 LATER_FRAMEWORK_VERSION_WARNING = 'This is not the latest supported version. ' \
                                   'If you would like to use version {latest}, ' \
                                   'please add framework_version={latest} to your constructor.'
+PYTHON_2_DEPRECATION_WARNING = 'The Python 2 {framework} images will be soon deprecated and may not be ' \
+                               'supported for newer upcoming versions of the {framework} images. ' \
+                               'Please set the argument \"py_version=\'py3\'\" to use the Python 3 {framework} image.'
+
 
 EMPTY_FRAMEWORK_VERSION_ERROR = 'framework_version is required for script mode estimator. ' \
                                 'Please add framework_version={} to your constructor to avoid this error.'
@@ -303,3 +307,7 @@ def empty_framework_version_warning(default_version, latest_version):
     if default_version != latest_version:
         msgs.append(LATER_FRAMEWORK_VERSION_WARNING.format(latest=latest_version))
     return ' '.join(msgs)
+
+
+def python_deprecation_warning(framework):
+    return PYTHON_2_DEPRECATION_WARNING.format(framework=framework)

src/sagemaker/mxnet/estimator.py

@@ -15,7 +15,8 @@
 import logging
 
 from sagemaker.estimator import Framework
-from sagemaker.fw_utils import framework_name_from_image, framework_version_from_tag, empty_framework_version_warning
+from sagemaker.fw_utils import framework_name_from_image, framework_version_from_tag, empty_framework_version_warning, \
+    python_deprecation_warning
 from sagemaker.mxnet.defaults import MXNET_VERSION
 from sagemaker.mxnet.model import MXNetModel
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
@@ -81,7 +82,7 @@ def __init__(self, entry_point, source_dir=None, hyperparameters=None, py_versio
                                     image_name=image_name, **kwargs)
 
         if py_version == 'py2':
-            logger.warning('mxnet py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
         self._configure_distribution(distributions)

src/sagemaker/mxnet/model.py

@@ -72,7 +72,7 @@ def __init__(self, model_data, role, entry_point, image=None, py_version='py2',
                                          **kwargs)
 
         if py_version == 'py2':
-            logger.warning('chainer py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
         self.framework_version = framework_version

src/sagemaker/pytorch/estimator.py

@@ -15,7 +15,8 @@
 import logging
 
 from sagemaker.estimator import Framework
-from sagemaker.fw_utils import framework_name_from_image, framework_version_from_tag, empty_framework_version_warning
+from sagemaker.fw_utils import framework_name_from_image, framework_version_from_tag, empty_framework_version_warning, \
+    python_deprecation_warning
 from sagemaker.pytorch.defaults import PYTORCH_VERSION, PYTHON_VERSION
 from sagemaker.pytorch.model import PyTorchModel
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
@@ -76,7 +77,7 @@ def __init__(self, entry_point, source_dir=None, hyperparameters=None, py_versio
         super(PyTorch, self).__init__(entry_point, source_dir, hyperparameters, image_name=image_name, **kwargs)
 
         if py_version == 'py2':
-            logger.warning('pytorch py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
 

src/sagemaker/pytorch/model.py

@@ -15,7 +15,7 @@
 import logging
 
 import sagemaker
-from sagemaker.fw_utils import create_image_uri, model_code_key_prefix
+from sagemaker.fw_utils import create_image_uri, model_code_key_prefix, python_deprecation_warning
 from sagemaker.model import FrameworkModel, MODEL_SERVER_WORKERS_PARAM_NAME
 from sagemaker.pytorch.defaults import PYTORCH_VERSION, PYTHON_VERSION
 from sagemaker.predictor import RealTimePredictor, npy_serializer, numpy_deserializer
@@ -72,7 +72,7 @@ def __init__(self, model_data, role, entry_point, image=None, py_version=PYTHON_
         super(PyTorchModel, self).__init__(model_data, image, role, entry_point, predictor_cls=predictor_cls, **kwargs)
 
         if py_version == 'py2':
-            logger.warning('pytorch py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
         self.framework_version = framework_version

src/sagemaker/sklearn/estimator.py

@@ -16,7 +16,7 @@
 
 from sagemaker.estimator import Framework
 from sagemaker.fw_registry import default_framework_uri
-from sagemaker.fw_utils import framework_name_from_image, empty_framework_version_warning
+from sagemaker.fw_utils import framework_name_from_image, empty_framework_version_warning, python_deprecation_warning
 from sagemaker.sklearn.defaults import SKLEARN_VERSION, SKLEARN_NAME
 from sagemaker.sklearn.model import SKLearnModel
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
@@ -79,6 +79,9 @@ def __init__(self, entry_point, framework_version=SKLEARN_VERSION, source_dir=No
         super(SKLearn, self).__init__(entry_point, source_dir, hyperparameters, image_name=image_name,
                                       **dict(kwargs, train_instance_count=1))
 
+        if py_version == 'py2':
+            logger.warning(python_deprecation_warning(self.__framework_name__))
+
         self.py_version = py_version
 
         if framework_version is None:

src/sagemaker/sklearn/model.py

@@ -12,13 +12,16 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
+import logging
+
 import sagemaker
-from sagemaker.fw_utils import model_code_key_prefix
+from sagemaker.fw_utils import model_code_key_prefix, python_deprecation_warning
 from sagemaker.fw_registry import default_framework_uri
 from sagemaker.model import FrameworkModel, MODEL_SERVER_WORKERS_PARAM_NAME
 from sagemaker.predictor import RealTimePredictor, npy_serializer, numpy_deserializer
 from sagemaker.sklearn.defaults import SKLEARN_VERSION, SKLEARN_NAME
 
+logger = logging.getLogger('sagemaker')
 
 class SKLearnPredictor(RealTimePredictor):
     """A RealTimePredictor for inference against Scikit-learn Endpoints.
@@ -68,6 +71,10 @@ def __init__(self, model_data, role, entry_point, image=None, py_version='py3',
         """
         super(SKLearnModel, self).__init__(model_data, image, role, entry_point, predictor_cls=predictor_cls,
                                            **kwargs)
+
+        if py_version == 'py2':
+            logger.warning(python_deprecation_warning(self.__framework_name__))
+
         self.py_version = py_version
         self.framework_version = framework_version
         self.model_server_workers = model_server_workers

src/sagemaker/tensorflow/estimator.py

@@ -29,7 +29,7 @@
 from sagemaker.utils import get_config_value
 from sagemaker.vpc_utils import VPC_CONFIG_DEFAULT
 
-LOGGER = logging.getLogger('sagemaker')
+logger = logging.getLogger('sagemaker')
 
 
 _FRAMEWORK_MODE_ARGS = ('training_steps', 'evaluation_steps', 'requirements_file', 'checkpoint_path')
@@ -231,14 +231,14 @@ def __init__(self, training_steps=None, evaluation_steps=None, checkpoint_path=N
             **kwargs: Additional kwargs passed to the Framework constructor.
         """
         if framework_version is None:
-            LOGGER.warning(fw.empty_framework_version_warning(TF_VERSION, self.LATEST_VERSION))
+            logger.warning(fw.empty_framework_version_warning(TF_VERSION, self.LATEST_VERSION))
         self.framework_version = framework_version or TF_VERSION
 
         super(TensorFlow, self).__init__(image_name=image_name, **kwargs)
         self.checkpoint_path = checkpoint_path
 
         if py_version == 'py2':
-            LOGGER.warning('tensorflow py2 container will be deprecated soon.')
+            logger.warning('tensorflow py2 container will be deprecated soon.')
 
         self.py_version = py_version
         self.training_steps = training_steps

src/sagemaker/tensorflow/model.py

@@ -15,13 +15,13 @@
 import logging
 
 import sagemaker
-from sagemaker.fw_utils import create_image_uri, model_code_key_prefix
+from sagemaker.fw_utils import create_image_uri, model_code_key_prefix, python_deprecation_warning
 from sagemaker.model import FrameworkModel, MODEL_SERVER_WORKERS_PARAM_NAME
 from sagemaker.predictor import RealTimePredictor
 from sagemaker.tensorflow.defaults import TF_VERSION
 from sagemaker.tensorflow.predictor import tf_json_serializer, tf_json_deserializer
 
-LOGGER = logging.getLogger('sagemaker')
+logger = logging.getLogger('sagemaker')
 
 
 class TensorFlowPredictor(RealTimePredictor):
@@ -73,7 +73,7 @@ def __init__(self, model_data, role, entry_point, image=None, py_version='py2',
                                               **kwargs)
 
         if py_version == 'py2':
-            LOGGER.warning('tensorflow py2 container will be deprecated soon.')
+            logger.warning(python_deprecation_warning(self.__framework_name__))
 
         self.py_version = py_version
         self.framework_version = framework_version