From b542c4b91704cbe60d2a0a2427a42449481e70b6 Mon Sep 17 00:00:00 2001 From: Joseph Keshet Date: Sat, 2 Jul 2016 07:25:50 +0300 Subject: [PATCH] fix interface --- helpers/utilities.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/helpers/utilities.py b/helpers/utilities.py index 8993c60..0b43356 100644 --- a/helpers/utilities.py +++ b/helpers/utilities.py @@ -54,9 +54,8 @@ def csv_append_row(tmp_preds, preds_filename, with_headers=True): f.write(line) - -def generate_tmp_filename(): - return tempfile._get_default_tempdir() + "/" + next(tempfile._get_candidate_names()) + ".txt" +def generate_tmp_filename(extension): + return tempfile._get_default_tempdir() + "/" + next(tempfile._get_candidate_names()) + "." + extension def logging_defaults(logging_level="INFO"):