From 2a7757ee7e061a15662d1064b5ff9ef6cfe97150 Mon Sep 17 00:00:00 2001 From: Abubakar Abid Date: Mon, 14 Mar 2022 09:37:30 -0500 Subject: [PATCH] formatting Former-commit-id: b78188d885230fd4ad617ee7288e840aa6dea321 --- gradio/flagging.py | 10 ++++++---- gradio/utils.py | 8 ++++---- test/test_utils.py | 3 +-- 3 files changed, 11 insertions(+), 10 deletions(-) diff --git a/gradio/flagging.py b/gradio/flagging.py index 89534973c9..2cf6ceb143 100644 --- a/gradio/flagging.py +++ b/gradio/flagging.py @@ -207,10 +207,12 @@ class CSVLogger(FlaggingCallback): writer.writerow(headers) writer.writerow(csv_data) with open(log_fp, "wb") as csvfile: - csvfile.write(utils.santize_for_csv( - encryptor.encrypt( - interface.encryption_key, output.getvalue().encode() - )) + csvfile.write( + utils.santize_for_csv( + encryptor.encrypt( + interface.encryption_key, output.getvalue().encode() + ) + ) ) else: if flag_index is None: diff --git a/gradio/utils.py b/gradio/utils.py index ef4ff3c2e9..ef6b083729 100644 --- a/gradio/utils.py +++ b/gradio/utils.py @@ -290,12 +290,13 @@ def get_default_args(func: Callable) -> Dict[str, Any]: def santize_for_csv(data: str | List[str] | List[List[str]]): - """ Sanitizes data so that it can be safely written to a CSV file. """ + """Sanitizes data so that it can be safely written to a CSV file.""" + def sanitize(item): return "'" + item - + unsafe_prefixes = ("+", "=", "-", "@") - + if isinstance(data, str): if data.startswith(unsafe_prefixes): warnings.warn("Sanitizing flagged data by escaping cell contents") @@ -318,4 +319,3 @@ def santize_for_csv(data: str | List[str] | List[List[str]]): return sanitized_data else: raise ValueError("Unsupported data type: " + str(type(data))) - diff --git a/test/test_utils.py b/test/test_utils.py index 75476360dd..eed645df62 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -15,8 +15,8 @@ from gradio.utils import ( json, launch_analytics, readme_to_html, - version_check, santize_for_csv, + version_check, ) os.environ["GRADIO_ANALYTICS_ENABLED"] = "False" @@ -135,6 +135,5 @@ class TestSanitizeForCSV(unittest.TestCase): self.assertEquals(safe_data, [["abc", "'=abc"]]) - if __name__ == "__main__": unittest.main()