small fixes in import error

Author: jiapinw

tests/integ/sagemaker/serve/constants.py

@@ -23,8 +23,8 @@
 SERVE_SAGEMAKER_ENDPOINT_TIMEOUT = 15
 SERVE_SAVE_TIMEOUT = 2
 
-PYTHON_VERSION_IS_38 = platform.python_version_tuple()[1] != "8"
-PYTHON_VERSION_IS_310 = platform.python_version_tuple()[1] != "10"
+PYTHON_VERSION_IS_NOT_38 = platform.python_version_tuple()[1] != "8"
+PYTHON_VERSION_IS_NOT_310 = platform.python_version_tuple()[1] != "10"
 
 XGB_RESOURCE_DIR = os.path.join(DATA_DIR, "serve_resources", "xgboost")
 PYTORCH_SQUEEZENET_RESOURCE_DIR = os.path.join(DATA_DIR, "serve_resources", "pytorch")

tests/integ/sagemaker/serve/test_serve_js_happy.py

@@ -17,7 +17,7 @@
 from sagemaker.serve.builder.schema_builder import SchemaBuilder
 from tests.integ.sagemaker.serve.constants import (
     SERVE_SAGEMAKER_ENDPOINT_TIMEOUT,
-    NOT_RUNNING_ON_PY310,
+    PYTHON_VERSION_IS_NOT_310,
 )
 
 from tests.integ.timeout import timeout
@@ -47,9 +47,10 @@ def happy_model_builder(sagemaker_session):
 
 
 @pytest.mark.skipif(
-    NOT_RUNNING_ON_PY310,
+    PYTHON_VERSION_IS_NOT_310,
     reason="The goal of these test are to test the serving components of our feature",
 )
+@pytest.mark.slow_test
 def test_happy_tgi_sagemaker_endpoint(happy_model_builder, gpu_instance_type):
     logger.info("Running in SAGEMAKER_ENDPOINT mode...")
     caught_ex = None

tests/integ/sagemaker/serve/test_serve_pt_happy.py

@@ -28,7 +28,7 @@
 from tests.integ.sagemaker.serve.constants import (
     PYTORCH_SQUEEZENET_RESOURCE_DIR,
     SERVE_SAGEMAKER_ENDPOINT_TIMEOUT,
-    PYTHON_VERSION_IS_310,
+    PYTHON_VERSION_IS_NOT_310,
 )
 from tests.integ.timeout import timeout
 from tests.integ.utils import cleanup_model_resources
@@ -149,12 +149,13 @@ def model_builder(request):
 
 
 @pytest.mark.skipif(
-    PYTHON_VERSION_IS_310,
+    PYTHON_VERSION_IS_NOT_310,
     reason="The goal of these test are to test the serving components of our feature",
 )
 @pytest.mark.parametrize(
     "model_builder", ["model_builder_inference_spec_schema_builder"], indirect=True
 )
+@pytest.mark.slow_test
 def test_happy_pytorch_local_container(sagemaker_session, model_builder, test_image):
     logger.info("Running in LOCAL_CONTAINER mode...")
     caught_ex = None
@@ -180,7 +181,7 @@ def test_happy_pytorch_local_container(sagemaker_session, model_builder, test_im
 
 
 @pytest.mark.skipif(
-    PYTHON_VERSION_IS_310,  # or NOT_RUNNING_ON_INF_EXP_DEV_PIPELINE,
+    PYTHON_VERSION_IS_NOT_310,  # or NOT_RUNNING_ON_INF_EXP_DEV_PIPELINE,
     reason="The goal of these test are to test the serving components of our feature",
 )
 @pytest.mark.parametrize(
@@ -222,7 +223,7 @@ def test_happy_pytorch_sagemaker_endpoint(
 
 
 # @pytest.mark.skipif(
-#     NOT_RUNNING_ON_INF_EXP_DEV_PIPELINE or PYTHON_VERSION_IS_310,
+#     NOT_RUNNING_ON_INF_EXP_DEV_PIPELINE or PYTHON_VERSION_IS_NOT_310,
 #     reason="The goal of these test are to test the serving components of our feature",
 # )
 # @pytest.mark.parametrize(
@@ -268,7 +269,7 @@ def test_happy_pytorch_sagemaker_endpoint(
 
 
 # @pytest.mark.skipif(
-#     NOT_RUNNING_ON_INF_EXP_DEV_PIPELINE or PYTHON_VERSION_IS_310,
+#     NOT_RUNNING_ON_INF_EXP_DEV_PIPELINE or PYTHON_VERSION_IS_NOT_310,
 #     reason="The goal of these test are to test the serving components of our feature",
 # )
 # @pytest.mark.parametrize(