This release is a pre-release and may not be stable for production use.
What is Kedro AzureML Pipeline?
Kedro AzureML Pipeline is a plugin that enables running Kedro pipelines on Azure ML Pipelines. It translates your Kedro pipeline into an Azure ML pipeline job where each Kedro node becomes a separate step.
Two deployment workflows are supported, both backed by Azure ML Environments:
- Code upload: only dependencies live in the Docker image; source code is uploaded at runtime (fast iteration for data scientists)
- Docker image: code is baked into the image (stable, repeatable workflows for MLOps)
Key features
| Feature | Description |
|---|---|
| Pipeline translation | Automatic Kedro node → Azure ML step mapping via the compile, run, and schedule CLI commands |
| Named jobs | Define multiple jobs in azureml.yml, each targeting a different pipeline, compute, or workspace |
| Scheduling | Attach cron or recurrence schedules to jobs for recurring Azure ML pipeline runs |
| Data assets | AzureMLAssetDataset for reading/writing Azure ML uri_file and uri_folder data assets |
| Distributed training | @distributed_job decorator with PyTorch, TensorFlow, and MPI backends |
| MLflow integration | Optional hook that wires Kedro-MLFlow to log under the correct Azure ML experiment |
| Multiple workspaces | Named workspace definitions with a __default__ fallback |
Installation
pip install kedro-azureml-pipeline
or with uv:
uv add kedro-azureml-pipeline
Quick start
1. Initialize configuration
kedro azureml init
This creates conf/base/azureml.yml with placeholder values and an .amlignore file.
2. Review the generated configuration
Open conf/base/azureml.yml and fill in your Azure details:
workspace:
__default__:
subscription_id: "<subscription_id>"
resource_group: "<resource_group>"
name: "<workspace_name>"
compute:
__default__:
cluster_name: "<cluster_name>"
execution:
environment: "<environment>"
code_directory: "."
3. Define a job and submit
Add a job to azureml.yml:
jobs:
training:
pipeline:
pipeline_name: "__default__"
experiment_name: "my-experiment"
Then submit it:
kedro azureml submit -j training
Use --dry-run to preview without submitting, or --wait-for-completion to block until the run finishes.
4. Compile to YAML (optional)
Export the Azure ML pipeline definition for inspection or CI:
kedro azureml compile -j training -o pipeline.yaml
Documentation
Full documentation is available at https://kedro-azureml-pipeline.readthedocs.io/.
Contributing
We welcome contributions, feedback, and questions:
- Report issues or request features: GitHub Issues
- Contributing guide: CONTRIBUTING.md
- Discussions: GitHub Discussions
License
This project is licensed under the terms of the Apache-2.0 License.
Acknowledgements
This project is a fork of kedro-azureml, originally developed by GetInData. We are grateful for their work in creating the initial plugin that bridges Kedro and Azure ML Pipelines. We have continued development to add new features, improve documentation, and maintain the project under the kedro-azureml-pipeline package name.
We would also like to thank Evolta Technologies for their support to the project.
This project is maintained by stateful-y, an ML consultancy specializing in MLOps and data science & engineering. If you're interested in collaborating or learning more about our services, please visit our website.
Metadata
Release files for kedro-azureml-pipeline 0.1.0a3
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| kedro_azureml_pipeline-0.1.0a3.tar.gz | 543.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| kedro_azureml_pipeline-0.1.0a3-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 601.1 kB
Release files / kedro_azureml_pipeline-0.1.0a3.tar.gz
| Download URL | kedro_azureml_pipeline-0.1.0a3.tar.gz |
|---|---|
| Size | 543.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
6a3473b37e6aa184bd8ab19525777828f0f52bf062ab7d31d2e718c6716e1de6
|
|
BLAKE2b-256 checksum How to use checksums |
ded21bf746e41c259879d6c89feee20219eead2d9a72718ea694d8bbc4749427
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Aug 12, 2026.
Transparency logRelease files / kedro_azureml_pipeline-0.1.0a3-py3-none-any.whl
| Download URL | kedro_azureml_pipeline-0.1.0a3-py3-none-any.whl |
|---|---|
| Size | 57.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
aa0679842380ac097379a3ddfacb6e176518e84ca2350458e1e6aa228f3c5c53
|
|
BLAKE2b-256 checksum How to use checksums |
7adadd004a8df5781fe1b143513420c5ce584dd73e4b403697da7963a82b4e1e
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Aug 12, 2026.
Transparency log