revert sm vllm (#5382)

Author: Jyothirmaikottu

src/image.py

@@ -130,9 +130,8 @@ def collect_installed_packages_information(self):
         ]
         for command in commands:
             command_responses.append(f"\n{command}")
-
             command_responses.append(
-                bytes.decode(docker_client.containers.run(self.ecr_url, command, entrypoint=""))
+                bytes.decode(docker_client.containers.run(self.ecr_url, command))
             )
         docker_client.containers.prune()
         return command_responses

src/start_testbuilds.py

@@ -311,11 +311,7 @@ def main():
 
             # Trigger sagemaker local test jobs when there are changes in sagemaker_tests
             # Skip SageMaker local tests for vLLM images as they use different test structure
-            if (
-                test_type == "sagemaker"
-                and config.is_sm_local_test_enabled()
-                and "vllm" not in images_str
-            ):
+            if test_type == "sagemaker" and config.is_sm_local_test_enabled():
                 test_job = f"dlc-pr-{test_type}-local-test"
                 run_test_job(commit, test_job, images_str)
 

test/testrunner.py

@@ -328,7 +328,6 @@ def main():
             "functionality_sanity",
             "security_sanity",
             "eks",
-            "sagemaker",
             "ec2",
         }:
             LOGGER.info(
@@ -398,7 +397,6 @@ def main():
         "bai",
         "quick_checks",
         "release_candidate_integration",
-        "sagemaker",
     ):
         pytest_rerun_arg = "--reruns=1"
         pytest_rerun_delay_arg = "--reruns-delay=10"
@@ -417,7 +415,7 @@ def main():
             pull_dlc_images(all_image_list)
         if specific_test_type == "bai":
             build_bai_docker_container()
-        if specific_test_type in ["eks", "ec2", "sagemaker"] and not is_all_images_list_eia:
+        if specific_test_type in ["eks", "ec2"] and not is_all_images_list_eia:
             frameworks_in_images = [
                 framework
                 for framework in ("mxnet", "pytorch", "tensorflow", "vllm")