Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
6570db2
1
Parent(s):
93ebdf4
Update process_data.py
Browse files- 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))
|