def _validate_restart_policy_argument()

in launcher/config_validator/value_validator.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

def _validate_restart_policy_argument(config: DictConfig) -> None:
    """
    Check only valid restartPolicy is provided if it is not None

    Parameters:
    config (DictConfig): Configuration dictionary
    """
    restart_policy_argument_name = "cluster.cluster_config.restartPolicy"
    restart_policy_argument = get_argument(config, restart_policy_argument_name)
    if restart_policy_argument is not None:
        supported_restart_policies = ["Always", "OnFailure", "Never", "ExitCode"]
        if restart_policy_argument not in supported_restart_policies:
            raise ValueError("Provided restartPolicy is not supported")