Fix static import check

TheCuriousWolf · Jul 21, 2022 · e71bd1a · e71bd1a
1 parent b92deee
commit e71bd1a
Show file tree

Hide file tree

Showing 2 changed files with 811 additions and 804 deletions.
diff --git a/notebooks/book2/17/bnn_hierarchical.ipynb b/notebooks/book2/17/bnn_hierarchical.ipynb
@@ -1028,7 +1028,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import flax.linen as nn\n",
+    "try:\n",
+    "    import flax.linen as nn\n",
+    "except ModuleNotFoundError:\n",
+    "    %pip install -qq flax\n",
+    "    import flax.linen as nn\n",
+    "\n",
     "from jax.flatten_util import ravel_pytree"
    ]
   },
@@ -1360,7 +1365,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.7"
+   "version": "3.9.12"
   }
  },
  "nbformat": 4,