def create_artifact_filter()

in src/sagemaker_huggingface_inference_toolkit/transformers_utils.py [0:0]

• 8 lines of code
• 2 McCabe index (conditional complexity)

def create_artifact_filter(framework):
    """
    Returns a list of regex pattern based on the DL Framework. which will be to used to ignore files when downloading
    """
    ignore_regex_list = list(set(FRAMEWORK_MAPPING.values()))

    pattern = FRAMEWORK_MAPPING.get(framework, None)
    if pattern in ignore_regex_list:
        ignore_regex_list.remove(pattern)
        return ignore_regex_list
    else:
        return []