Gagan Bhatia commited on
Commit
255f974
1 Parent(s): fc0bcb4

Update process_data.py

Browse files
Files changed (1) hide show
  1. src/data/process_data.py +1 -0
src/data/process_data.py CHANGED
@@ -12,6 +12,7 @@ def process_data(split='train'):
12
  df.columns = ['Unnamed: 0', 'input_text', 'output_text']
13
  df = df.sample(frac=params['split'], replace=True, random_state=1)
14
  if os.path.exists("data/raw/{}.csv".format(split)):
 
15
  df.to_csv('data/processed/{}.csv'.format(split))
16
 
17
 
 
12
  df.columns = ['Unnamed: 0', 'input_text', 'output_text']
13
  df = df.sample(frac=params['split'], replace=True, random_state=1)
14
  if os.path.exists("data/raw/{}.csv".format(split)):
15
+ os.remove("data/raw/{}.csv".format(split))
16
  df.to_csv('data/processed/{}.csv'.format(split))
17
 
18