change config files

ServerlessLLM · Feb 20, 2025 · 83fe0f1 · 83fe0f1
1 parent bc52f16
commit 83fe0f1
Show file tree

Hide file tree

Showing 5 changed files with 20 additions and 14 deletions.
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/...m/datasets_config/datasets_config_hf.json → examples/datasets_config_hf.json b/...m/datasets_config/datasets_config_hf.json → 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/...atasets_config/datasets_config_local.json → examples/datasets_config_local.json b/...atasets_config/datasets_config_local.json → 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