Improve _save_file method to handle both dict and str inputs (#1011)

Improve _save_file method to handle both dict and str inputs (#1011) #326