fix import

Author: jinyan-li1

test/testrunner.py

@@ -3,6 +3,7 @@
 import sys
 import logging
 import re
+import importlib.util
 
 from multiprocessing import Pool, Manager
 from datetime import datetime
@@ -33,7 +34,6 @@
 from test_utils import KEYS_TO_DESTROY_FILE
 from test_utils.pytest_cache import PytestCache
 from test.vllm.trigger_test import test as test_vllm
-from .infra.test.entrypoint import main as run_new_tests
 
 from src.codebuild_environment import get_codebuild_project_name
 
@@ -437,6 +437,15 @@ def main():
                 try:
                     LOGGER.info(f"Running vLLM EKS EC2 tests with image: {all_image_list[0]}")
                     if new_test_structure_enabled:
+                        spec = importlib.util.spec_from_file_location(
+                            "entrypoint",
+                            os.path.join(
+                                os.path.dirname(__file__), "..", ".infra", "test", "entrypoint.py"
+                            ),
+                        )
+                        entrypoint_module = importlib.util.module_from_spec(spec)
+                        spec.loader.exec_module(entrypoint_module)
+                        run_new_tests = entrypoint_module.main
                         LOGGER.info("Using new buildspec-based test system")
                         run_new_tests()
                     else: