Gagan Bhatia commited on
Commit
6570db2
1 Parent(s): 93ebdf4

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
@@ -8,6 +8,7 @@ def process_data(split="train"):
8
  with open("params.yml") as f:
9
  params = yaml.safe_load(f)
10
 
 
11
  if os.path.exists("data/raw/{}.csv".format(split)):
12
  os.remove("data/raw/{}.csv".format(split))
13
  df.to_csv('data/processed/{}.csv'.format(split))
 
8
  with open("params.yml") as f:
9
  params = yaml.safe_load(f)
10
 
11
+ df = pd.read_csv("data/raw/{}.csv".format(split))
12
  if os.path.exists("data/raw/{}.csv".format(split)):
13
  os.remove("data/raw/{}.csv".format(split))
14
  df.to_csv('data/processed/{}.csv'.format(split))