Update demo_choice_set.ipynb

transp-or · Oct 15, 2023 · 97c4d60 · 97c4d60
1 parent 725a277
commit 97c4d60
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/demo_notebooks/demo_choice_set.ipynb b/demo_notebooks/demo_choice_set.ipynb
@@ -41,12 +41,12 @@
     "import joblib\n",
     "import pandas as pd\n",
     "import numpy as np\n",
+    "import oasisabm as oa\n",
+
     "\n",
-    "import sys\n",
-    "sys.path.append('estimation/')\n",
-    "\n",
-    "from estimation import ChoiceSetGenerator\n",
-    "from helper_func import parse_df_schedule, schedule_to_pandas\n"
+
+    "from oa.estimation import ChoiceSetGenerator\n",
+    "from oa.helper_func import parse_df_schedule, schedule_to_pandas\n"
    ]
   },
   {