Update process_data.py

src/data/process_data.py

@@ -13,3 +13,4 @@ if __name__ == "__main__":
     with open("data_params.yml") as f:
         params = yaml.safe_load(f)
 
+    process_data(frac=params["split"], split="train")