Add torch

geometric-intelligence · May 24, 2024 · 2d5632d · 2d5632d
1 parent 2216945
commit 2d5632d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tutorials/02_methods_estimate_manifold_dimension.ipynb b/tutorials/02_methods_estimate_manifold_dimension.ipynb
@@ -88,6 +88,7 @@
     "    skdim_dimension_estimation,\n",
     ")\n",
     "\n",
+    "import torch\n",
     "os.environ[\"GEOMSTATS_BACKEND\"] = \"pytorch\"\n",
     "import geomstats.backend as gs"
    ]