Skip to content

T
tic
Commit 5eaf74ab by Neo Chien Committed by Tianqi Chen
Options

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

* Not assuming HOME in tvm/download.py

* Trigger notification
parent 137bf5f4
Showing with 4 additions and 1 deletions
python/tvm/contrib/download.py
...@@ -119,7 +119,10 @@ def download(url, path, overwrite=False, size_compare=False, verbose=1, retries= ...@@ -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 '')) .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) os.makedirs(TEST_DATA_ROOT_PATH, exist_ok=True)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment