Azure / Azure/azure-sdk-for-python

Data tab not getting populated as part of the Model Register

Open
#37,962 3 comments 0 reactions 1 assignee Assigned to @achauhan-scc View on GitHub
Client customer-reported feature-request Machine Learning needs-team-attention Service Attention
Dominant language
Python
Stars
5.6k
Forks
3.4k
Avg merge
1d 21h
Merged PRs (30d)
193

Description

- **Package Name**: azure.ai.ml.entities
- **Package Version**: v2

**Describe the bug**

Follow up issues as part of the bug: [36551](https://github.com/Azure/azure-sdk-for-python/issues/36551)
I tried registering the model with the below: https://learn.microsoft.com/en-us/azure/machine-learning/how-to-manage-models?view=azureml-api-2&tabs=python

run_model = Model(
path="azureml://jobs/{id}/outputs/model_folder/",
name="test1-v2-example",
description="Model created from run.",
type=AssetTypes.CUSTOM_MODEL,
)
ml_client.models.create_or_update(run_model)

This did register the Model, **But it did not associate the Data Asset Tab for the Registered Model**

It seems like the UI expects the data to be populated there or at least should have the ability to list that info?

**Why is this important?**

Because customers want to be able to easily see what job and what data was used to produce a given model registered in the registry---in case there are questions on model performance during production inference for example. Observability on data lineage and code lineage to produce a model is critical to give decision makers confidence that a model was trained correctly on the right data. Thanks!

**Screenshots**
![Image](https://github.com/user-attachments/assets/06e8bd9b-411a-4faf-bc9f-e897908d9d56)

**Additional context**
Add any other context about the problem here.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.