Merge pull request #109 from MichaelSkralivetsky/main

model_endpoints_tags_labels/test-ml-6609.ipynb: fix
mlrun · Jan 8, 2025 · f6f72b3 · f6f72b3
2 parents 2f59d60 + 4e133fe
commit f6f72b3
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/model_endpoints_tags_labels/test-ml-6609.ipynb b/model_endpoints_tags_labels/test-ml-6609.ipynb
@@ -231,7 +231,9 @@
    "id": "c6decf5d-4349-4bed-9a17-9ddf7c10ae61",
    "metadata": {},
    "outputs": [],
-   "source": "end_point = db.list_model_endpoints(project=project.name, model_name=\"model\", model_tag=f\"v{i}\").endpoints[0]"
+   "source": [
+    "end_point = db.list_model_endpoints(project=project.name, model_name=\"model\", model_tag=f\"v{i}\").endpoints[0]"
+   ]
   },
   {
    "cell_type": "markdown",
@@ -260,7 +262,7 @@
    "outputs": [],
    "source": [
     "#test model endpoint tag\n",
-    "assert end_point.spec.model.split(\":\")[-1]==f\"v{i}\""
+    "assert end_point.spec.model_tag == f\"v{i}\""
    ]
   },
   {
@@ -437,7 +439,9 @@
    "id": "6ea1fd4e-ff31-4807-add5-9005f3b2d011",
    "metadata": {},
    "outputs": [],
-   "source": "end_point = db.list_model_endpoints(project=project.name, model_name=\"model\", model_tag=f\"v{i}\").endpoints[0]"
+   "source": [
+    "end_point = db.list_model_endpoints(project=project.name, model_name=\"model\", model_tag=f\"v{i}\").endpoints[0]"
+   ]
   },
   {
    "cell_type": "markdown",
@@ -466,7 +470,7 @@
    "outputs": [],
    "source": [
     "# test model endpoint tags\n",
-    "assert end_point.spec.model.split(\":\")[-1]==f\"v{i}\""
+    "assert end_point.spec.model_tag == f\"v{i}\""
    ]
   },
   {