From 0df4e62da00f5c5b990b4dfe3dc40650e0c13965 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Sun, 29 Dec 2019 10:06:50 -0500 Subject: [PATCH] [http] Tweak http user-agent (#2353) --- src/transformers/file_utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/transformers/file_utils.py b/src/transformers/file_utils.py index 2e0153f46c..60cc9baa6c 100644 --- a/src/transformers/file_utils.py +++ b/src/transformers/file_utils.py @@ -98,7 +98,6 @@ def is_torch_available(): def is_tf_available(): - return _tf_available @@ -274,6 +273,10 @@ def s3_get(url, temp_file, proxies=None): def http_get(url, temp_file, proxies=None, resume_size=0, user_agent=None): ua = "transformers/{}; python/{}".format(__version__, sys.version.split()[0]) + if is_torch_available(): + ua += "; torch/{}".format(torch.__version__) + if is_tf_available(): + ua += "; tensorflow/{}".format(tf.__version__) if isinstance(user_agent, dict): ua += "; " + "; ".join("{}/{}".format(k, v) for k, v in user_agent.items()) elif isinstance(user_agent, str):