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