Azure / azure-sdk-for-python

This repository is for active development of the Azure SDK for Python. For consumers of the SDK we recommend visiting our public developer docs at https://learn.microsoft.com/python/azure/ or our versioned developer docs at https://azure.github.io/azure-sdk-for-python.
MIT License
4.63k stars 2.84k forks source link

Data tab not getting populated as part of the Model Register #37962

Open sweanan opened 1 month ago

sweanan commented 1 month ago

Describe the bug

Follow up issues as part of the bug: 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

Additional context Add any other context about the problem here.

github-actions[bot] commented 1 month ago

Thanks for the feedback! We are routing this to the appropriate team for follow-up. cc @Azure/azure-ml-sdk @azureml-github.