add_index_to_datasets

asreview · Sep 18, 2024 · f6a6cb1 · f6a6cb1
1 parent a39c570
commit f6a6cb1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/asreviewcontrib/makita/template_prior.py b/asreviewcontrib/makita/template_prior.py
@@ -141,9 +141,13 @@ def get_template_specific_params(self, params):
         print(f"Total rows of non-prior knowledge: {total_rows_without_priors}")
 
         # Save the combined dataset to the appropriate file paths
-        combined_dataset.to_csv(filepath_with_priors, index=False)
+        combined_dataset.to_csv(filepath_with_priors, 
+                                index=True, 
+                                index_label='record_id')
         combined_dataset[combined_dataset["makita_priors"] != 1].to_csv(
-            filepath_without_priors, index=False
+            filepath_without_priors, 
+            index=True,
+            index_label='record_id'
         )
 
         # Create a string of indices for rows with prior knowledge