updates

docs/tutorials/notebooks/Unbalanced_OT.ipynb

@@ -21,6 +21,19 @@
     "Instead of $\\rho_a$ and $\\rho_b$, `OTT`'s solver uses parameters `tau_a`$=\\rho_a /(\\varepsilon+ \\rho_a)$ and `tau_b`$=\\rho_b /(\\varepsilon+ \\rho_b)$. Setting either of these parameters to 1 corresponds to setting the corresponding $\\rho_a, \\rho_b$ to $\\infty$ and solving the corresponding balanced {class}`~ott.problems.linear.linear_problem.LinearProblem`. "
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "06f20970",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import sys\n",
+    "\n",
+    "if \"google.colab\" in sys.modules:\n",
+    "    !pip install -q git+https://github.com/ott-jax/ott@main"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 1,
@@ -41,11 +54,11 @@
     "\n",
     "import matplotlib.pyplot as plt\n",
     "\n",
-    "from ott.geometry import costs, pointcloud\n",
+    "from ott.geometry import pointcloud\n",
     "from ott.problems.linear import linear_problem\n",
     "from ott.solvers.linear import sinkhorn\n",
     "from ott.tools import plot\n",
-    "from ott.tools.gaussian_mixture import gaussian_mixture, probabilities"
+    "from ott.tools.gaussian_mixture import gaussian_mixture"
    ]
   },
   {