change config files

XuehengWang · XuehengWang · commit 83fe0f10fbd2 · 2025-02-20T18:29:03.000Z
diff --git a/examples/datasets_config_default.json b/examples/datasets_config_default.json
@@ -0,0 +1,13 @@
+{
+    "sort_strategy": "random",
+    "dataset_1": {
+        "file_name": "",
+        "prompt_field": "",
+        "select_ratio": 1
+    },
+    "dataset_2": {
+        "file_name": "",
+        "prompt_field": "",
+        "select_ratio": 1
+    }
+}
diff --git a/examples/datasets_config_hf.json b/examples/datasets_config_hf.json
@@ -1,5 +1,5 @@
 {
-    "sort": "original",
+    "sort_strategy": "original",
     "dataset_1": {
         "file_name": "hf://datasets/fka/awesome-chatgpt-prompts/prompts.csv",
         "prompt_field": "prompt",
diff --git a/examples/datasets_config_local.json b/examples/datasets_config_local.json
@@ -1,12 +1,12 @@
 {
-    "sort": "random",
+    "sort_strategy": "random",
     "dataset_1": {
         "file_name": "conversation_sample.csv",
         "prompt_field": "user_message",
         "select_ratio": 6
     },
     "dataset_2": {
-        "file_name": "../datasets/GPT4_coding_sample.csv",
+        "file_name": "~/.cache/tracestorm/GPT4_coding_sample.csv",
         "prompt_field": "user_prompt",
         "select_ratio": 4
     }
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,4 +43,7 @@ ignore = ["B007"]  # Loop control variable not used within loop body
 
 [tool.isort]
 use_parentheses = true
-skip_gitignore = true
+skip_gitignore = true
+
+[tool.setuptools]
+packages = { find = { exclude = ["examples"] } }
diff --git a/tracestorm/datasets_config/datasets_config_missing.json b/tracestorm/datasets_config/datasets_config_missing.json

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,5 @@`
`1`	`1`	`{`
`2`		`- "sort": "original",`
	`2`	`+ "sort_strategy": "original",`
`3`	`3`	`"dataset_1": {`
`4`	`4`	`"file_name": "hf://datasets/fka/awesome-chatgpt-prompts/prompts.csv",`
`5`	`5`	`"prompt_field": "prompt",`