change: include more notebook tests, logger to warn (#1953)

Author: metrizable

src/sagemaker/deprecations.py

@@ -17,6 +17,9 @@
 import warnings
 
 logging.captureWarnings(True)
+logger = logging.getLogger(__name__)
+
+V2_URL = "https://sagemaker.readthedocs.io/en/stable/v2.html"
 
 
 def _warn(msg):
@@ -25,12 +28,9 @@ def _warn(msg):
     Args:
         phrase: The phrase to include in the warning.
     """
-    warnings.warn(
-        f"{msg} in sagemaker>=2.\n"
-        "See: https://sagemaker.readthedocs.io/en/stable/v2.html for details.",
-        DeprecationWarning,
-        stacklevel=2,
-    )
+    full_msg = f"{msg} in sagemaker>=2.\nSee: {V2_URL} for details."
+    warnings.warn(full_msg, DeprecationWarning, stacklevel=2)
+    logger.warning(full_msg)
 
 
 def removed_warning(phrase):

src/sagemaker/predictor.py

@@ -19,6 +19,7 @@
     deprecated_serialize,
     removed_kwargs,
     renamed_kwargs,
+    renamed_warning,
 )
 from sagemaker.deserializers import (  # noqa: F401 # pylint: disable=unused-import
     BytesDeserializer,
@@ -412,6 +413,12 @@ def accept(self):
         """The content type(s) that are expected from the inference endpoint."""
         return self.deserializer.ACCEPT
 
+    @property
+    def endpoint(self):
+        """Deprecated attribute. Please use endpoint_name."""
+        renamed_warning("The endpoint attribute")
+        return self.endpoint_name
+
 
 csv_serializer = deprecated_serialize(CSVSerializer(), "csv_serializer")
 json_serializer = deprecated_serialize(JSONSerializer(), "json_serializer")

tests/scripts/run-notebook-test.sh

@@ -22,12 +22,15 @@ echo "set SAGEMAKER_ROLE_ARN=$SAGEMAKER_ROLE_ARN"
 ./amazon-sagemaker-examples/advanced_functionality/kmeans_bring_your_own_model/kmeans_bring_your_own_model.ipynb \
 ./amazon-sagemaker-examples/advanced_functionality/tensorflow_iris_byom/tensorflow_BYOM_iris.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/1P_kmeans_highlevel/kmeans_mnist.ipynb \
+./amazon-sagemaker-examples/sagemaker-python-sdk/1P_kmeans_lowlevel/kmeans_mnist_lowlevel.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/managed_spot_training_mxnet/managed_spot_training_mxnet.ipynb \
+./amazon-sagemaker-examples/sagemaker-python-sdk/mxnet_gluon_sentiment/mxnet_sentiment_analysis_with_gluon.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/mxnet_onnx_eia/mxnet_onnx_eia.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/mxnet_onnx_export/mxnet_onnx_export.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/pytorch_cnn_cifar10/pytorch_local_mode_cifar10.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/scikit_learn_randomforest/Sklearn_on_SageMaker_end2end.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/tensorflow_moving_from_framework_mode_to_script_mode/tensorflow_moving_from_framework_mode_to_script_mode.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/tensorflow_script_mode_pipe_mode/tensorflow_script_mode_pipe_mode.ipynb \
+./amazon-sagemaker-examples/sagemaker-python-sdk/tensorflow_script_mode_quickstart/tensorflow_script_mode_quickstart.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/tensorflow_script_mode_using_shell_commands/tensorflow_script_mode_using_shell_commands.ipynb \
 ./amazon-sagemaker-examples/sagemaker-python-sdk/tensorflow_serving_using_elastic_inference_with_your_own_model/tensorflow_serving_pretrained_model_elastic_inference.ipynb