diff --git a/registry/import_models_to_verta/mlflow_databricks_bulk_import_example.ipynb b/registry/import_models_to_verta/mlflow_databricks_bulk_import_example.ipynb index 27760f7..6b9d77f 100644 --- a/registry/import_models_to_verta/mlflow_databricks_bulk_import_example.ipynb +++ b/registry/import_models_to_verta/mlflow_databricks_bulk_import_example.ipynb @@ -118,7 +118,7 @@ "source": [ "mlflow_models: List[MLflow_RegisteredModel] = list()\n", "\n", - "result: PagedList = mlflow_client.search_registered_models()\n", + "result: PagedList = mlflow_client.search_registered_models(max_results=10, order_by=["name ASC"])\n", "mlflow_models += result.to_list()\n", "while result.token: # handle paginated results\n", " result: PagedList = mlflow_client.search_registered_models(page_token=result.token)\n",