[fix] notebook descriptions

Author: kierandidi

notebooks/adding_new_dataset_tutorial.ipynb

@@ -352,7 +352,7 @@
     "cfg = config.validate_config(cfg)\n",
     "\n",
     "datamodule = hydra.utils.instantiate(cfg.dataset.datamodule)\n",
-    "datamodule.setup(\"train\")\n",
+    "datamodule.setup(\"fit\")\n",
     "dl = datamodule.train_dataloader()\n",
     "\n",
     "for i in dl:\n",
@@ -448,7 +448,7 @@
     "cfg = config.validate_config(cfg)\n",
     "\n",
     "datamodule = hydra.utils.instantiate(cfg)\n",
-    "datamodule.setup(\"train\")\n",
+    "datamodule.setup(\"fit\")\n",
     "dl = datamodule.train_dataloader()\n",
     "\n",
     "for i in dl:\n",

notebooks/customizing_existing_dataset_tutorial.ipynb

@@ -84,7 +84,7 @@
     "    overrides=[\n",
     "        \"encoder=schnet\",\n",
     "        \"task=inverse_folding\",\n",
-    "        \"dataset=cath\",\n",
+    "        \"dataset=pdb\",\n",
     "        \"features=ca_angles\",\n",
     "        \"+aux_task=none\",\n",
     "    ],\n",
@@ -118,7 +118,7 @@
     "cfg = config.validate_config(cfg)\n",
     "\n",
     "datamodule = hydra.utils.instantiate(cfg.dataset.datamodule)\n",
-    "datamodule.setup(\"train\")\n",
+    "datamodule.setup(\"fit\")\n",
     "dl = datamodule.train_dataloader()\n",
     "\n",
     "for i in dl:\n",
@@ -207,7 +207,7 @@
     "cfg = config.validate_config(cfg)\n",
     "\n",
     "datamodule = hydra.utils.instantiate(cfg)\n",
-    "datamodule.setup(\"train\")\n",
+    "datamodule.setup(\"fit\")\n",
     "dl = datamodule.train_dataloader()\n",
     "\n",
     "for i in dl:\n",