Azure / Azure/azure-sdk-for-python
Data tab not getting populated as part of the Model Register
- 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**

**Additional context**
Add any other context about the problem here.
Contributor guide
Assessment
This issue has not been assessed yet.