add spider config and comment

eosphoros-ai · Jan 1, 2024 · 77f9ead · 77f9ead
1 parent f2122b3
commit 77f9ead
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/dbgpt_hub/configs/config.py b/dbgpt_hub/configs/config.py
@@ -55,12 +55,13 @@
         "db_id_name": "db_id",
         "output_name": "SQL",
         "is_multiple_turn": False,
-    },
+    }
     # {
     #     "data_source": "spider",
     #     "train_file": ["train_spider.json", "train_others.json"],
     #     "dev_file": ["dev.json"],
-    #     "tables_file": "tables.json",
+    #     "train_tables_file": "tables.json",
+    #     "dev_tables_file": "tables.json",
     #     "db_id_name": "db_id",
     #     "output_name": "query",
     #     "is_multiple_turn": False,

diff --git a/dbgpt_hub/data_process/sql_data_process.py b/dbgpt_hub/data_process/sql_data_process.py
@@ -70,6 +70,7 @@ def decode_json_file(
                                     + " is the primary key."
                                     + "\n"
                             )
+                    # combination primary key
                     elif type(primary_key[j]) == list:
                         combine_p = "The combination of ("
                         keys = []