Skip to main content
Pre-release

This release is a pre-release and may not be stable for production use.

Kedro AzureML Pipeline

Python Version License PyPI Version codecov

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:

License

This project is licensed under the terms of the Apache-2.0 License.

How do I cite Kedro AzureML Pipeline?

If you use Kedro AzureML Pipeline in work you publish, please cite it:

Guillaume Tauzin. Kedro AzureML Pipeline: Orchestrate Kedro pipelines with Azure ML Pipeline. https://github.com/stateful-y/kedro-azureml-pipeline

Or in BibTeX:

@software{kedro_azureml_pipeline,
  author  = "Guillaume Tauzin",
  title   = "{Kedro AzureML Pipeline: Orchestrate Kedro pipelines with Azure ML Pipeline}",
  url     = "https://github.com/stateful-y/kedro-azureml-pipeline",
  license = "Apache-2.0"
}

Reference managers can read CITATION.cff directly. To cite a specific version, see the citation page.

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.


Evolta Technologies


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.

Made by stateful-y

Metadata

Release files for kedro-azureml-pipeline 0.1.0a4

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for kedro-azureml-pipeline 0.1.0a4
File Size Uploaded
kedro_azureml_pipeline-0.1.0a4.tar.gz 685.9 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for kedro-azureml-pipeline 0.1.0a4
File Interpreter ABI Platform
kedro_azureml_pipeline-0.1.0a4-py3-none-any.whl Python 3 none any Details

Total release size: 765.3 kB

Release files / kedro_azureml_pipeline-0.1.0a4.tar.gz

Download URL kedro_azureml_pipeline-0.1.0a4.tar.gz
Size 685.9 kB
Tags Source
SHA-256 checksum
How to use checksums
11d0f53b8787b6ded7d28d6ed2155f3d933c8af2fce3cf6c3e7ecd5ee12750ee
BLAKE2b-256 checksum
How to use checksums
dc497d7185be4cb189734aaf8202a7f81b2edf2609731ff8d788f6de2921ab16
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 Oct 19, 2026.

Transparency log

Release files / kedro_azureml_pipeline-0.1.0a4-py3-none-any.whl

Download URL kedro_azureml_pipeline-0.1.0a4-py3-none-any.whl
Size 79.4 kB
Tags Python 3
SHA-256 checksum
How to use checksums
3fff5edf3568da3eff1c3c8bc3dadd3e93ab3ad34f852c81bd6b397c26063e0f
BLAKE2b-256 checksum
How to use checksums
7bb6caad48187fa9904b23c7d86624c0e705d829f90468c2ae9c1deda3e2ee47
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 Oct 19, 2026.

Transparency log
Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page