src/transformers/integrations/integration_utils.py/AzureMLCallback/__init__
class AzureMLCallback:    def __init__(self, azureml_run=None):
        if not is_azureml_available():
            raise RuntimeError("AzureMLCallback requires azureml to be installed. Run `pip install azureml-sdk`.")
        self.azureml_run = azureml_run