Spaces:

librarian-bots
/

dataset-column-search-api

Runtime error

davanstrien HF staff commited on Jul 17, 2024

Commit

2bb8e35

verified ·

1 Parent(s): 0089a9c

Update create_collections.py

Files changed (1) hide show

create_collections.py CHANGED Viewed

@@ -106,7 +106,7 @@ def update_collection_for_dataset(
 collections = [
     {
         "dataset_columns": ["chosen", "rejected", "prompt"],
-        "collection_description": "Datasets suitable for Direct Preference Optimization based on having 'chosen', 'rejected', and 'prompt' columns",
         "collection_name": "Direct Preference Optimization Datasets",
     },
     {

 collections = [
     {
         "dataset_columns": ["chosen", "rejected", "prompt"],
+        "collection_description": "Datasets suitable for DPO based on having 'chosen', 'rejected', and 'prompt' columns. Created using librarian-bots/dataset-column-search-api",
         "collection_name": "Direct Preference Optimization Datasets",
     },
     {