fix dataset forum export

p2m2 · Oct 22, 2024 · 4f7a89a · 4f7a89a
1 parent e8d799a
commit 4f7a89a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/llm_semantic_annotator/abstract/abstract_manager.py b/llm_semantic_annotator/abstract/abstract_manager.py
@@ -246,7 +246,8 @@ def build_dataset_abstracts_annotations(self):
                         'pmid' : pmid_list,
                         'reference_id' : reference_id_list
                     })
-                    outf = self.config['retention_dir']+f"/QueryResultEntry_{abstracts_gen}.csv"
-                    print(outf)
-                    df.to_csv(outf, index=False)
+                    if not df.empty:
+                        outf = self.config['retention_dir']+f"/QueryResultEntry_{abstracts_gen}.csv"
+                        print(outf)
+                        df.to_csv(outf, index=False)