fix: redundant export of not gzipped pandas Dataframe removed.

This commit is contained in:
Fabian Joswig 2022-06-30 15:49:40 +01:00
parent 153cc795b8
commit c6ec11045c

View file

@ -30,10 +30,12 @@ def dump_df(df, fname, gz=True):
if not fname.endswith('.csv'):
fname += '.csv'
out.to_csv(fname, index=False)
if gz is True:
with open(fname, 'rb') as f_in, gzip.open(fname + ".gz", 'wb') as f_out:
f_out.writelines(f_in)
if not fname.endswith('.gz'):
fname += '.gz'
out.to_csv(fname, index=False, compression='gzip')
else:
out.to_csv(fname, index=False)
def load_df(fname, auto_gamma=False, gz=True):