Add dataroot to prediction tutorial, see nutonomy#382 (nutonomy#384)

cawcawlabs · May 5, 2020 · be551d8 · be551d8
1 parent 3211ed8
commit be551d8
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python-sdk/tutorials/prediction_tutorial.ipynb b/python-sdk/tutorials/prediction_tutorial.ipynb
@@ -16,7 +16,7 @@
     "\n",
     "It is organized into the following sections:\n",
     "\n",
-    "1. Data splis for the challenge\n",
+    "1. Data splits for the challenge\n",
     "2. Getting past and future data for an agent \n",
     "3. Changes to the Map API\n",
     "4. Overview of input representation\n",
@@ -33,7 +33,7 @@
     "from nuscenes import NuScenes\n",
     "\n",
     "# This is the path where you stored your copy of the nuScenes dataset.\n",
-    "DATAROOT = '/data/sets/nuscenes/'\n",
+    "DATAROOT = '/data/sets/nuscenes'\n",
     "\n",
     "nuscenes = NuScenes('v1.0-mini', dataroot=DATAROOT)"
    ]
@@ -58,7 +58,7 @@
    "outputs": [],
    "source": [
     "from nuscenes.eval.prediction.splits import get_prediction_challenge_split\n",
-    "mini_train = get_prediction_challenge_split(\"mini_train\")\n",
+    "mini_train = get_prediction_challenge_split(\"mini_train\", dataroot=DATAROOT)\n",
     "mini_train[:5]"
    ]
   },
@@ -210,7 +210,7 @@
    "outputs": [],
    "source": [
     "from nuscenes.map_expansion.map_api import NuScenesMap\n",
-    "nusc_map = NuScenesMap(map_name='singapore-onenorth')"
+    "nusc_map = NuScenesMap(map_name='singapore-onenorth', dataroot=DATAROOT)"
    ]
   },
   {
@@ -690,7 +690,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.4"
+   "version": "3.7.7"
   }
  },
  "nbformat": 4,