def is_test_job_implemented_for_framework()

in src/start_testbuilds.py [0:0]

• 40 lines of code
• 11 McCabe index (conditional complexity)

def is_test_job_implemented_for_framework(images_str, test_type):
    """
    Check to see if a test job is implemented and supposed to be executed for this particular set of images
    """
    is_trcomp_image = False
    is_huggingface_trcomp_image = False
    is_huggingface_image = False
    if "huggingface" in images_str:
        if "trcomp" in images_str:
            is_huggingface_trcomp_image = True
        else:
            is_huggingface_image = True
    elif "trcomp" in images_str:
        is_trcomp_image = True

    is_autogluon_image = "autogluon" in images_str

    if (is_huggingface_image or is_autogluon_image) and test_type in [
        constants.EC2_TESTS,
        constants.EC2_BENCHMARK_TESTS,
        constants.ECS_TESTS,
        constants.EKS_TESTS,
    ]:
        LOGGER.debug(f"Skipping {test_type} test")
        return False
        # SM Training Compiler has EC2 tests implemented so don't skip
    if is_huggingface_trcomp_image and (
        test_type
        in [
            constants.ECS_TESTS,
            constants.EKS_TESTS,
            constants.EC2_BENCHMARK_TESTS,
        ]
    ):
        LOGGER.debug(f"Skipping {test_type} tests for huggingface trcomp containers")
        return False

    if is_trcomp_image and (
        test_type
        in [
            constants.EKS_TESTS,
            constants.EC2_BENCHMARK_TESTS,
        ]
    ):
        LOGGER.debug(f"Skipping {test_type} tests for trcomp containers")
        return False
    return True