add secrets to yml

argilla-io · May 8, 2024 · a6477a0 · a6477a0
1 parent 681e16e
commit a6477a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -74,11 +74,11 @@ def build_parsed_results(dataset):
         results = rg.FeedbackDataset.from_argilla(RESULTS_DATASET, workspace=RESULTS_WORKSPACE)
         results.add_records(completed_local_records)
         if DELETE_SOURCE_RECORDS:
-            dataset.delete_records([r.id for r in completed_remote_records])
+            dataset.delete_records([r for r in completed_remote_records])
     except Exception as e:
         results = local_submitted.push_to_argilla(RESULTS_DATASET, workspace=RESULTS_WORKSPACE)
         if DELETE_SOURCE_RECORDS:
-            dataset.delete_records([r.id for r in completed_remote_records])
+            dataset.delete_records([r for r in completed_remote_records])
 
     parsed_results_dataset = build_parsed_results(results)
     print(f"Pushing dataset to {PARSED_RESULTS_DATASET}...")