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
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") == ""