def apply_all_patches()

in ultravox/utils/monkey_patches.py [0:0]


def apply_all_patches():
    """
    Apply all patches at once.
    """
    global IS_PATCHED
    if IS_PATCHED:
        logger.info("Patches already applied, skipping")
        return

    logger.info("Starting to apply patches...")

    # Patch HF Hub methods
    patch_with_retry(huggingface_hub.HfApi, "dataset_info")
    patch_with_retry(huggingface_hub.HfApi, "model_info")
    patch_with_retry(huggingface_hub.HfApi, "repo_info")

    # Patch http_backoff
    patch_hf_hub_http_backoff()

    # Patch audio decoder
    patch_audio_decoder()

    # Patch datasets methods
    patch_with_retry(
        datasets.load.HubDatasetModuleFactoryWithParquetExport, "get_module"
    )

    IS_PATCHED = True
    logger.info("Successfully applied all patches")