Merge pull request #613 from aviaIguazio/1.6.x-dev

[mask detection] fixing 6984
mlrun · Jun 30, 2024 · e1380d0 · e1380d0
2 parents b561532 + ee13359
commit e1380d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mask-detection/1-training-and-evaluation.ipynb b/mask-detection/1-training-and-evaluation.ipynb
@@ -108,7 +108,7 @@
    "outputs": [],
    "source": [
     "if framework=='tf-keras':\n",
-    "    commands = ['pip install tensorflow~=2.9.0',\n",
+    "    commands = ['pip install tensorflow~=2.9.0 mlrun',\n",
     "                'pip install horovod==0.25.0']\n",
     "    builder_env = {'HOROVOD_WITH_MPI':'1', 'HOROVOD_WITH_TENSORFLOW':'1'}\n",
     "else:\n",