ezermoysis
New Contributor III

‎01-09-2025 08:52 AM

Thanks @Alberto_Umana. I think so, yes.

I am using the following code to register the models and when I print the

# Use Unity Catalog as the model registry
mlflow.set_registry_uri("databricks-uc")
print(model_info.model_uri)
# Register the model to the Unity Catalog
uc_registered_model_info = mlflow.register_model(
model_uri=model_info.model_uri, name=UC_MODEL_NAME
)

and when I print

print(uc_registered_model_info)

I get this, so I believe there are versions registered

<ModelVersion: aliases=[], creation_timestamp=1736436606155, current_stage=None, description='', last_updated_timestamp=1736436613763, name='emoysis_catalog.cookbook.sop_chatbot', run_id='', run_link=None, source='dbfs:/databricks/mlflow-tracking/3498969875739103/0ee0f2039ef54180b474e285380fabca/artifacts/agent', status='READY', status_message='', tags={}, user_id='e.moysis@instadeep.com', version='10'>

 

0 Kudos