Pulastya B commited on
Commit
c84c9dd
·
1 Parent(s): a126ee5

Fix auto_ml_pipeline: ensure output_path has .csv extension

Browse files
Files changed (1) hide show
  1. src/tools/auto_pipeline.py +4 -0
src/tools/auto_pipeline.py CHANGED
@@ -56,6 +56,10 @@ def auto_ml_pipeline(file_path: str,
56
  if output_path is None:
57
  output_path = "./outputs/data/auto_pipeline_output.csv"
58
 
 
 
 
 
59
  Path(output_path).parent.mkdir(parents=True, exist_ok=True)
60
 
61
  results = {
 
56
  if output_path is None:
57
  output_path = "./outputs/data/auto_pipeline_output.csv"
58
 
59
+ # Ensure output_path has .csv extension
60
+ if not output_path.endswith('.csv'):
61
+ output_path = output_path.rstrip('/\\') + '.csv'
62
+
63
  Path(output_path).parent.mkdir(parents=True, exist_ok=True)
64
 
65
  results = {