modified data reading file

classical/aml-cli-v2/data-science/src/prep.py

@@ -71,11 +71,7 @@ def main(args):
     # ------------ Reading Data ------------ #
     # -------------------------------------- #
 
-    print("mounted_path files: ")
-    arr = os.listdir(args.raw_data)
-    print(arr)
-
-    data = pd.read_csv((Path(args.raw_data) / 'taxi-data.csv'))
+    data = pd.read_csv((Path(args.raw_data)))
     data = data[NUMERIC_COLS + CAT_NOM_COLS + CAT_ORD_COLS + [TARGET_COL]]
 
     # ------------- Split Data ------------- #

classical/rai-aml-cli-v2/data-science/src/prep/prep.py

@@ -71,11 +71,7 @@ def main(args):
     # ------------ Reading Data ------------ #
     # -------------------------------------- #
 
-    print("mounted_path files: ")
-    arr = os.listdir(args.raw_data)
-    print(arr)
-
-    data = pd.read_csv((Path(args.raw_data) / 'taxi-data.csv'))
+    data = pd.read_csv((Path(args.raw_data)))
     data = data[NUMERIC_COLS + CAT_NOM_COLS + CAT_ORD_COLS + [TARGET_COL]]
 
     # ------------- Split Data ------------- #