Fix CSV writer on Windows

PiperOrigin-RevId: 325892185
Change-Id: I96053328785323a40450e4b33704d873a6c6f38b
This commit is contained in:
Austin Anderson 2020-08-10 14:47:36 -07:00 committed by TensorFlower Gardener
parent e477d16d52
commit 6411114cbd

View File

@ -364,8 +364,9 @@ def main():
print("DRY RUN: Generated this TSV row:")
print("\t".join(map(str, next_tsv_row)))
else:
with open("data.tsv", "w") as tsvfile:
writer = csv.writer(tsvfile, delimiter="\t", quoting=csv.QUOTE_MINIMAL)
with open("data.tsv", "w", newline="") as tsvfile:
writer = csv.writer(tsvfile, delimiter="\t", quoting=csv.QUOTE_MINIMAL,
lineterminator=os.linesep)
writer.writerow(next_tsv_row)
bq([
"load", "--source_format", "CSV", "--field_delimiter", "tab",