Unit-test: link to the default downloading dir. #131

test/test_opennre_bert_infer.py

@@ -14,6 +14,7 @@
 from arelight.predict.writer_csv import TsvPredictWriter
 from arelight.run.utils import OPENNRE_CHECKPOINTS
 from arelight.third_party.torch import sentence_re_loader
+from arelight.utils import get_default_download_dir
 
 
 class TestLoadModel(unittest.TestCase):
@@ -44,7 +45,7 @@ def infer_bert(pretrain_path, labels_scaler, output_file_gzip, predefined, ckpt_
  model = BertOpenNREInferencePipelineItem.init_bert_model(
  pretrain_path=pretrain_path, labels_scaler=labels_scaler, ckpt_path=ckpt_path,
  device_type="cpu", max_length=max_length, mask_entity=mask_entity,
- dir_to_donwload=os.getcwd(), pooler=pooler, predefined=predefined)
+ dir_to_donwload=get_default_download_dir(), pooler=pooler, predefined=predefined)
 
  eval_loader = sentence_re_loader(path=test_data_file,
  table_name="contents",