Merge pull request #107 from MichaelSkralivetsky/main

test-mlflow-tracking.ipynb: improve assert
mlrun · Dec 3, 2024 · f20aa4e · f20aa4e
2 parents 1c466c7 + 0f0d9b5
commit f20aa4e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test-mlflow-tracking/test-mlflow-tracking.ipynb b/test-mlflow-tracking/test-mlflow-tracking.ipynb
@@ -574,9 +574,9 @@
     "artifacts = project.list_artifacts()\n",
     "assert artifacts\n",
     "assert len(artifacts)==3\n",
-    "assert artifacts[0]['metadata']['key']=='model'\n",
-    "assert artifacts[1]['metadata']['key']=='feature_importance_weight_png'\n",
-    "assert artifacts[2]['metadata']['key']=='feature_importance_weight_json'\n",
+    "artifacts = {artifacts[0]['metadata']['key'], artifacts[1]['metadata']['key'], artifacts[2]['metadata']['key']}\n",
+    "expected_artifacts = {'feature_importance_weight_png', 'feature_importance_weight_json', 'model'}\n",
+    "assert artifacts == expected_artifacts, f\"expected: {expected_artifacts}, received: {artifacts}\"\n",
     "\n",
     "models = project.list_models()\n",
     "assert models\n",