Skip to content

Added scheduled GitHub Action to monitor a batch endpoint via MLflow - #1

Open
Chakradhar886 wants to merge 1 commit into
mainfrom
v-chakradhar/batch-endpoint-monitor
Open

Added scheduled GitHub Action to monitor a batch endpoint via MLflow#1
Chakradhar886 wants to merge 1 commit into
mainfrom
v-chakradhar/batch-endpoint-monitor

Conversation

@Chakradhar886

Copy link
Copy Markdown
Owner

Summary

Adds a GitHub Action that periodically checks an Azure ML batch endpoint's recent runs and fails the workflow if any run is in a non-healthy state.

Changes

  • .github/workflows/monitor-batch.yml - cron (every 30 min) + workflow_dispatch trigger; wires four repo secrets into env vars and runs the monitor script.
  • tutorials/batch_endpoint_monitor/monitor.py - uses azure-ai-ml + mlflow to list runs for the endpoint experiment and raises if any run status is not FINISHED or RUNNING.
  • tutorials/batch_endpoint_monitor/requirements.txt - azure-ai-ml, azure-identity, mlflow, azureml-mlflow.

Required repo secrets

  • SUBSCRIPTION_ID
  • RESOURCE_GROUP
  • WORKSPACE_NAME
  • BATCH_ENDPOINT_NAME

Notes

  • DefaultAzureCredential is used; for unattended CI you'll typically wire OIDC / federated credentials or a service principal.
  • Draft - opening for review before enabling the schedule.

- .github/workflows/monitor-batch.yml: cron + workflow_dispatch trigger; reads four secrets and runs the monitor script.
- tutorials/batch_endpoint_monitor/monitor.py: uses azure-ai-ml + mlflow to list runs for the endpoint experiment and fails the job if any run is not FINISHED/RUNNING.
- tutorials/batch_endpoint_monitor/requirements.txt: azure-ai-ml, azure-identity, mlflow, azureml-mlflow.
@Chakradhar886
Chakradhar886 marked this pull request as ready for review June 11, 2026 09:57
@Chakradhar886 Chakradhar886 changed the title Add scheduled GitHub Action to monitor a batch endpoint via MLflow Added scheduled GitHub Action to monitor a batch endpoint via MLflow Jun 11, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant