output_path = f"file:///home/vignesh/Saravana/bundle_apps/final_file/"

(
    input_df
    .repartition(1)
    .write
    .format("csv")
    .option("header", "true")
    .partitionBy("device_type")
    .option("delimiter", ",")
    .mode("overwrite")
    .save(output_path)
)