style: pre-commit fixes

Author: pre-commit-ci[bot]

docs/src/pytree.ipynb

@@ -115,7 +115,7 @@
     "        y0=flat_y0,\n",
     "        t_eval=t_eval,\n",
     "    )\n",
-    "    return pytree.unflatten(treedef, flat_solution.y)\n"
+    "    return pytree.unflatten(treedef, flat_solution.y)"
    ]
   },
   {
@@ -153,7 +153,7 @@
     "        y=20 * u.m / u.s,\n",
     "    )\n",
     "    * object_mass\n",
-    ")\n"
+    ")"
    ]
   },
   {
@@ -183,7 +183,7 @@
     "    position, momentum = state\n",
     "    dposition_dt = momentum / object_mass\n",
     "    dmomentum_dt = force(position, momentum)\n",
-    "    return (dposition_dt, dmomentum_dt)\n"
+    "    return (dposition_dt, dmomentum_dt)"
    ]
   },
   {
@@ -258,7 +258,7 @@
     "ax.set_xlabel(\"x (m)\")\n",
     "ax.set_ylabel(\"y (m)\")\n",
     "ax.set_title(\"Projectile motion\")\n",
-    "ax.legend()\n"
+    "ax.legend()"
    ]
   },
   {