From 97c4d600ee34fffe85257b1592e4881429f9f2ec Mon Sep 17 00:00:00 2001
From: Janody <44189326+Janody@users.noreply.github.com>
Date: Sun, 15 Oct 2023 21:41:44 +0200
Subject: [PATCH] Update demo_choice_set.ipynb

---
 demo_notebooks/demo_choice_set.ipynb | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/demo_notebooks/demo_choice_set.ipynb b/demo_notebooks/demo_choice_set.ipynb
index fc0b437..d0ff4a1 100644
--- 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"
    ]
   },
   {