Skip to main content

Python bindings for the Windows AI Machine Learning EP Catalog API

Project description

windowsml

Python bindings for the Windows AI Machine Learning API.

This package provides a ctypes-based wrapper around the WinML flat C API, enabling Python applications to discover and manage execution providers and model catalogs for ONNX Runtime on Windows.

Installation

pip install windowsml

ONNX Runtime dependency

To automatically install a matching version of ONNX Runtime, use the with-ort extra:

pip install windowsml[with-ort]

If you choose not to use [with-ort], you must install a compatible version of onnxruntime separately and ensure it matches the version expected by this package.

Quick Start

EP Catalog — discover execution providers

from windowsml import EpCatalog

with EpCatalog() as catalog:
    for ep in catalog.find_all_providers():
        print(f"{ep.name} v{ep.version} - {ep.ready_state.name}")

        # Ensure the provider is ready (downloads/installs if needed)
        ep.ensure_ready()

Model Catalog — discover and download models

from windowsml import ModelCatalogSource, ModelCatalog

with ModelCatalogSource.create_from_uri("https://example.com/catalog.json") as source:
    with ModelCatalog([source]) as catalog:
        for model in catalog.find_all_models():
            print(f"{model.name} v{model.version} by {model.publisher}")
            print(f"  Size: {model.model_size_in_bytes} bytes")
            print(f"  EPs:  {model.execution_providers}")

API Reference

Enums

  • EpReadyStateReady, NotReady, NotPresent
  • EpCertificationUnknown, Certified, Uncertified
  • CatalogModelStatusReady, NotReady
  • CatalogModelInstanceStatusAvailable, InProgress, Unavailable

Classes

EpCatalog()

Context manager for the EP catalog.

Method Description
find_all_providers() Returns list[ExecutionProvider]
close() Release the catalog

ExecutionProvider

Handle to a discovered EP (owned by the catalog).

Property / Method Type
name str
version str
package_family_name str
library_path str
package_root_path str
ready_state EpReadyState
certification EpCertification
ensure_ready() Blocks until ready
ensure_ready_async(on_complete, on_progress) Returns AsyncOperation

ModelCatalogSource

Represents a catalog source endpoint. Create via static factory methods.

Method / Property Description
create_from_uri(uri) Create from a URI or local path (blocking)
create_from_uri_async(uri, ...) Create from a URI (async)
create_from_uri_with_headers(uri, headers) Create with custom HTTP headers (blocking)
create_from_uri_with_headers_async(uri, headers, ...) Create with custom HTTP headers (async)
id str — source identifier
uri str — source URI
close() Release the source

ModelCatalog(sources)

Context manager for the model catalog. Created from a list of ModelCatalogSource.

Method / Property Description
execution_providers list[str] — configured EPs
set_execution_providers(providers) Set EP filter for queries
get_available_model(id_or_name) Returns CatalogModelInfo
get_available_models() Returns list[CatalogModelInfo]
find_model(id_or_name) Find a model (blocking)
find_model_async(id_or_name, ...) Find a model (async)
find_all_models() Find all models (blocking)
find_all_models_async(...) Find all models (async)
close() Release the catalog

CatalogModelInfo

Metadata about a model in the catalog.

Property / Method Type
id str
name str
publisher str
source_id str
version str
license str
license_uri str
license_text str
uri str
model_size_in_bytes int
execution_providers list[str]
status CatalogModelStatus
get_instance_async(...) Returns AsyncOperationCatalogModelInstance
get_instance_with_headers_async(headers, ...) Returns AsyncOperationCatalogModelInstance
close() Release the model info

CatalogModelInstance

A downloaded or local instance of a catalog model.

Property / Method Type
model_paths list[str] — local file paths
model_info CatalogModelInfo
close() Release the instance

AsyncOperation

Returned by async methods. Supports context-manager usage.

Method Description
wait() Block until complete
get_status(wait=False) Poll or wait for status
get_result() Extract the typed result (if applicable)
cancel() Request cancellation
close() Release async resources

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

windowsml-2.0.297rc0-py3-none-win_arm64.whl (407.5 kB view details)

Uploaded Python 3Windows ARM64

windowsml-2.0.297rc0-py3-none-win_amd64.whl (411.9 kB view details)

Uploaded Python 3Windows x86-64

File details

Details for the file windowsml-2.0.297rc0-py3-none-win_arm64.whl.

File metadata

File hashes

Hashes for windowsml-2.0.297rc0-py3-none-win_arm64.whl
Algorithm Hash digest
SHA256 88c93c13317072b4c577e446b759b968f0e0f21d3d3ebdf5ae683592c621ff96
MD5 9282b37e60cc0adc3952a20a100407a6
BLAKE2b-256 ad89297d8d73db4e5bf24a8dc782bad2a7b9b54d2295106f8d1d813c1924abe7

See more details on using hashes here.

File details

Details for the file windowsml-2.0.297rc0-py3-none-win_amd64.whl.

File metadata

File hashes

Hashes for windowsml-2.0.297rc0-py3-none-win_amd64.whl
Algorithm Hash digest
SHA256 451fb6f0eeb8e21a3ad3ba3e51a89506c4416c6a722cea9c8f44a4caf424d807
MD5 7129b92a9094e98a3fcd86da85097a2d
BLAKE2b-256 fdd3d5e052bef90d22340163d7695250aa368ad46001ddeee6c92fa2e409ef50

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page