Beispiel #1
0
def test_developer_configuration():
    """
    Ensure that defaults are set back to normal before merge
    """
    # Check dev settings
    assert config.parse_dlc_developer_configs("dev", "partner_developer") == ""
    assert config.parse_dlc_developer_configs("dev", "ei_mode") is False
    assert config.parse_dlc_developer_configs("dev", "neuron_mode") is False
    assert config.parse_dlc_developer_configs("dev", "graviton_mode") is False
    assert config.parse_dlc_developer_configs("dev", "benchmark_mode") is False
    assert config.parse_dlc_developer_configs("dev", "habana_mode") is False

    # Check build settings
    assert config.parse_dlc_developer_configs("build", "skip_frameworks") == []
    assert config.parse_dlc_developer_configs("build", "datetime_tag") is True
    assert config.parse_dlc_developer_configs("build", "do_build") is True

    # Check test settings
    assert config.parse_dlc_developer_configs("test", "sanity_tests") is True
    assert config.parse_dlc_developer_configs("test", "sagemaker_remote_tests") == "off"
    assert config.parse_dlc_developer_configs("test", "sagemaker_local_tests") is False
    assert config.parse_dlc_developer_configs("test", "ecs_tests") is True
    assert config.parse_dlc_developer_configs("test", "eks_tests") is True
    assert config.parse_dlc_developer_configs("test", "ec2_tests") is True
    assert config.parse_dlc_developer_configs("test", "use_scheduler") is False
    assert config.parse_dlc_developer_configs("test", "safety_check_test") is False
    assert config.parse_dlc_developer_configs("test", "ecr_scan_allowlist_feature") is False
Beispiel #2
0
def test_build_version_override_configuration():
    """
    Ensure that buildspec override defaults are set back to normal before merge
    """
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-autogluon") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-mxnet") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-pytorch") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-1") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-2") == ""

    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-huggingface-pytorch") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-huggingface-tensorflow") == ""

    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-pytorch-habana") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-2-habana") == ""

    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-mxnet-neuron") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-pytorch-neuron") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-1-neuron") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-2-neuron") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-huggingface-pytorch-neuron") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-huggingface-tensorflow-neuron") == ""

    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-huggingface-pytorch-trcomp") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-huggingface-tensorflow-2-trcomp") == ""

    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-mxnet-graviton") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-pytorch-graviton") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-2-graviton") == ""

    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-mxnet-eia") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-pytorch-eia") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-1-eia") == ""
    assert config.parse_dlc_developer_configs("buildspec_override", "dlc-pr-tensorflow-2-eia") == ""