fix typo in docs

BuddenD · Jun 9, 2020 · d3ccf0a · d3ccf0a
1 parent 307701c
commit d3ccf0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/notebooks/How_JAX_primitives_work.ipynb b/docs/notebooks/How_JAX_primitives_work.ipynb
@@ -1280,7 +1280,7 @@
         "  if not ad.is_undefined_primal(x):\n",
         "    # This use of multiply_add is with a constant \"x\"\n",
         "    assert ad.is_undefined_primal(y)\n",
-        "    ct_y = ad.Zero(y.aval) if type(ct) is ad.Zero else multiply_add_prim(x, ct, lax.zeros_like_array(x))
+        "    ct_y = ad.Zero(y.aval) if type(ct) is ad.Zero else multiply_add_prim(x, ct, lax.zeros_like_array(x))\n",
         "    res = None, ct_y, ct\n",
         "  else:\n",
         "    # This use of multiply_add is with a constant \"y\"\n",