Merge pull request #154 from george0st/change

Small tuning
george0st · Apr 18, 2024 · 6398032 · 6398032
2 parents a04188a + 3749e98
commit 6398032
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/docs/todo_list.md b/docs/todo_list.md
@@ -2,5 +2,4 @@
 
 The list of expected/future improvements:
 
-0. Extend unit tests (if header exists)
 1. Add ability to select type of output format (CSV or parquet) in commandline)
diff --git a/generator/base_data.py b/generator/base_data.py
@@ -59,7 +59,7 @@ def save(self, path, dir: str, compress: bool):
 
         # write CSV
         output_csv = os.path.join(path, f"{self.name}.csv.gz" if compress else f"{self.name}.csv")
-        append_csv=True if os.path.isfile(output_csv) else False
+        append_csv=False if (self._parquet_writer is None) else True
         df.to_csv(output_csv,
                   header=False if append_csv else True,
                   index=False,
Original file line number	Diff line number	Diff line change
Expand Up		@@ -2,5 +2,4 @@

		The list of expected/future improvements:

		0. Extend unit tests (if header exists)
		1. Add ability to select type of output format (CSV or parquet) in commandline)