Add default cache dir

yzhangcs · yzhangcs · commit a3962b25b278 · 2022-05-23T09:35:13.000+08:00
diff --git a/supar/utils/common.py b/supar/utils/common.py
@@ -1,8 +1,12 @@
 # -*- coding: utf-8 -*-
 
+import os
+
 PAD = '<pad>'
 UNK = '<unk>'
 BOS = '<bos>'
 EOS = '<eos>'
 
 MIN = -1e32
+
+CACHE = os.path.expanduser('~/.cache/supar')
diff --git a/supar/utils/fn.py b/supar/utils/fn.py
@@ -7,6 +7,7 @@
 import zipfile
 
 import torch
+from supar.utils.common import CACHE
 
 
 def ispunct(token):
@@ -238,7 +239,7 @@ def pad(tensors, padding_value=0, total_length=None, padding_side='right'):
 
 
 def download(url, reload=False):
-    path = os.path.join(os.path.expanduser('~/.cache/supar'), os.path.basename(urllib.parse.urlparse(url).path))
+    path = os.path.join(CACHE, os.path.basename(urllib.parse.urlparse(url).path))
     os.makedirs(os.path.dirname(path), exist_ok=True)
     if reload:
         os.remove(path) if os.path.exists(path) else None
