Merge pull request #31 from derVedro/main

fix double write line to csv
This commit is contained in:
pharmapsychotic
2023-03-25 12:55:23 -05:00
committed by GitHub

View File

@@ -42,7 +42,6 @@ class BatchWriter:
elif self.mode == BATCH_OUTPUT_MODES[1]:
self.file.write(f"{prompt}\n")
elif self.mode == BATCH_OUTPUT_MODES[2]:
self.file.write(f"{file},{prompt}\n")
self.csv.writerow([file, prompt])
def close(self):