Merge pull request #614 from mlrun/1.6.x-dev

[mask detection] fixing cannot import name 'builder' from 'google.protobuf.internal'
mlrun · Jun 30, 2024 · 2ccf80c · 2ccf80c
2 parents c5a4a45 + e1380d0
commit 2ccf80c
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",