Merge pull request #312 from pyt-team/frantzen/fix-scnn-notebook

Fix dataset loading in tutorials
pyt-team · Jan 13, 2025 · 359d62e · 359d62e
2 parents 637fcbb + e5252fb
commit 359d62e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tutorials/simplicial/scnn_train.ipynb b/tutorials/simplicial/scnn_train.ipynb
@@ -104,8 +104,6 @@
    "source": [
     "shrec, _ = tnx.datasets.shrec_16(size=\"small\")\n",
     "\n",
-    "shrec = {key: np.array(value) for key, value in shrec.items()}\n",
-    "\n",
     "x_0s = shrec[\"node_feat\"]\n",
     "x_1s = shrec[\"edge_feat\"]\n",
     "x_2s = shrec[\"face_feat\"]\n",
@@ -900,7 +898,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3 (ipykernel)",
+   "display_name": ".venv",
    "language": "python",
    "name": "python3"
   },
@@ -914,7 +912,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.3"
+   "version": "3.12.7"
   }
  },
  "nbformat": 4,