def _get_rule_list()

in smdebug_rulesconfig/debugger_rules/_utils.py [0:0]

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

def _get_rule_list(framework):
    framework = framework.upper()
    rule_set = UNIVERSAL_RULES
    if framework in SUPPORTED_DL_FRAMEWORKS:
        rule_set = rule_set.union(DEEP_LEARNING_RULES).union(DEEP_LEARNING_APPLICATION_RULES)
    elif framework == "XGBOOST":
        rule_set = rule_set.union(XGBOOST_RULES)
    else:
        raise Exception(f"{framework} is not supported by debugger rules")

    return list(rule_set)