[TEST] Not assuming HOME in tvm/download.py (#3803)

* Not assuming HOME in tvm/download.py

* Trigger notification

diff --git a/python/tvm/contrib/download.py b/python/tvm/contrib/download.py
index b3ba7ef..2dd9559 100644 (file)
--- a/python/tvm/contrib/download.py
+++ b/python/tvm/contrib/download.py
@@ -119,7 +119,10 @@ def download(url, path, overwrite=False, size_compare=False, verbose=1, retries=
                       .format(repr(err), retries, 's' if retries > 1 else ''))
 
 
-TEST_DATA_ROOT_PATH = os.path.join(os.path.expanduser('~'), '.tvm_test_data')
+if "TEST_DATA_ROOT_PATH" in os.environ:
+    TEST_DATA_ROOT_PATH = os.environ.get("TEST_DATA_ROOT_PATH")
+else:
+    TEST_DATA_ROOT_PATH = os.path.join(os.path.expanduser('~'), '.tvm_test_data')
 os.makedirs(TEST_DATA_ROOT_PATH, exist_ok=True)