Adds asyncio patch to MNIST tutorial.

PiperOrigin-RevId: 257660361
google-parfait · Jul 11, 2019 · a13d07c · a13d07c
1 parent 197ce96
commit a13d07c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/docs/tutorials/federated_learning_for_image_classification.ipynb b/docs/tutorials/federated_learning_for_image_classification.ipynb
@@ -124,7 +124,12 @@
         "# NOTE: The high-performance executor components used in this tutorial are not\n",
         "# yet included in the released pip package; you may need to compile from source.\n",
         "!pip install --quiet tensorflow_federated\n",
-        "!pip install --quiet  tf-nightly"
+        "!pip install --quiet  tf-nightly\n",
+        "\n",
+        "# NOTE: Jupyter requires a patch to asyncio.\n",
+        "!pip install --upgrade nest_asyncio\n",
+        "import nest_asyncio\n",
+        "nest_asyncio.apply()"
       ]
     },
     {