Modify path to relative path

test_pretrained.ipynb

@@ -48,7 +48,7 @@
     "    hugging_face_path = snapshot_download(\n",
     "        repo_id=\"USC-Applied-NLP-Group/SQL-Generation\",\n",
     "        repo_type=\"model\",  \n",
-    "        allow_patterns=[\"src/*\", \"train-data/*\", \"deepseek-coder-1.3b-instruct/*\"], \n",
+    "        allow_patterns=[\"src/*\", \"train-data/*\", \"deepseek-coder-1.3b-instruct/*\", \"nba-data/*\"], \n",
     "    )\n",
     "    sys.path.append(hugging_face_path)\n",
     "    current_path = hugging_face_path"
@@ -166,7 +166,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 29,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -179,7 +179,7 @@
    ],
    "source": [
     "# Create connection to sqlite3 database\n",
-    "connection = sql.connect('./nba-data/nba.sqlite')\n",
+    "connection = sql.connect(get_path('nba-data/nba.sqlite'))\n",
     "cursor = connection.cursor()\n",
     "\n",
     "# Execute query from model output and print result\n",
@@ -304,7 +304,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -325,7 +325,7 @@
     }
    ],
    "source": [
-    "less_than_90_df = pd.read_csv(\"./train-data/less_than_90.tsv\", sep='\\t')\n",
+    "less_than_90_df = pd.read_csv(get_path(\"train-data/less_than_90.tsv\"), sep='\\t')\n",
     "run_evaluation(less_than_90_df, \"Less than 90\")\n",
     "print(\"Dataset length: \" + str(len(less_than_90_df)))"
    ]
@@ -339,7 +339,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -372,7 +372,7 @@
     }
    ],
    "source": [
-    "game_queries = pd.read_csv(\"./train-data/queries_from_game.tsv\", sep='\\t')\n",
+    "game_queries = pd.read_csv(get_path(\"train-data/queries_from_game.tsv\"), sep='\\t')\n",
     "run_evaluation(game_queries, \"Queries from game\")\n",
     "print(\"Dataset length: \" + str(len(game_queries)))"
    ]
@@ -386,7 +386,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -406,7 +406,7 @@
     }
    ],
    "source": [
-    "other_stats_queries = pd.read_csv(\"./train-data/queries_from_other_stats.tsv\", sep='\\t')\n",
+    "other_stats_queries = pd.read_csv(get_path(\"train-data/queries_from_other_stats.tsv\"), sep='\\t')\n",
     "run_evaluation(other_stats_queries, \"Queries from other stats\")\n",
     "print(\"Dataset length: \" + str(len(other_stats_queries)))"
    ]
@@ -420,7 +420,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -438,7 +438,7 @@
     }
    ],
    "source": [
-    "team_queries = pd.read_csv(\"./train-data/queries_from_team.tsv\", sep='\\t')\n",
+    "team_queries = pd.read_csv(get_path(\"train-data/queries_from_team.tsv\"), sep='\\t')\n",
     "run_evaluation(team_queries, \"Queries from team\")\n",
     "print(\"Dataset length: \" + str(len(team_queries)))"
    ]
@@ -452,7 +452,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 13,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -472,7 +472,7 @@
     }
    ],
    "source": [
-    "join_queries = pd.read_csv(\"./train-data/with_join.tsv\", sep='\\t')\n",
+    "join_queries = pd.read_csv(get_path(\"train-data/with_join.tsv\"), sep='\\t')\n",
     "run_evaluation(join_queries, \"Queries with join\")\n",
     "print(\"Dataset length: \" + str(len(join_queries)))"
    ]
@@ -486,7 +486,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 14,
+   "execution_count": null,
    "metadata": {},
    "outputs": [
     {
@@ -520,7 +520,7 @@
     }
    ],
    "source": [
-    "no_join_queries = pd.read_csv(\"./train-data/without_join.tsv\", sep='\\t')\n",
+    "no_join_queries = pd.read_csv(get_path(\"train-data/without_join.tsv\"), sep='\\t')\n",
     "run_evaluation(no_join_queries, \"Queries without join\")\n",
     "print(\"Dataset length: \" + str(len(no_join_queries)))"
    ]