update

Author: xnuohz

test/llm/models/test_txt2kg.py

@@ -83,42 +83,3 @@ def fake_llm_fn(chunk, **kwargs):
     assert isinstance(out, list)
     assert all(len(t) == 3 for t in out)
     os.remove(out_path)
-
-
-# @onlyOnline
-# @withPackage("transformers")
-# def test_chunk_to_triples_str_local_no_external_import():
-#     class FakeLLM:
-#         def __init__(self, name):
-#             self.name = name
-
-#         def eval(self):
-#             return self
-
-#         def inference(self, question, max_tokens):
-#             return ["('A','B','C')"]
-
-#     fake_module = types.SimpleNamespace(LLM=FakeLLM)
-#     sys.modules["torch_geometric.nn.nlp"] = fake_module
-
-#     model = TXT2KG(local_LM=True)
-#     out = model._chunk_to_triples_str_local("This is a test text.")
-#     assert isinstance(out, str)
-#     assert "A" in out
-#     assert model.total_chars_parsed > 0
-#     assert model.time_to_parse > 0
-
-# @onlyOnline
-# @withPackage("transformers")
-# def test_add_doc_and_save():
-#     model = TXT2KG(local_LM=True)
-
-#     model.add_doc_2_KG("graph knowledge text", QA_pair=("Q1", "A1"))
-#     assert ("Q1", "A1") in model.relevant_triples
-#     assert model.relevant_triples[("Q1", "A1")] == [("a", "b", "c")]
-
-#     # 保存 + 加载
-#     save_path = "/tmp/kg.pt"
-#     model.save_kg(save_path)
-#     loaded = torch.load(save_path)
-#     assert loaded["Q1", "A1"] == [("a", "b", "c")]

test/nn/test_model_hub.py

@@ -83,13 +83,13 @@ def test_save_pretrained_with_push_to_hub(model, tmp_path):
     )
 
 
-# @withPackage('huggingface_hub')
-# def test_from_pretrained(model, tmp_path):
-#     save_directory = f'{str(tmp_path / REPO_NAME)}'
-#     model.save_pretrained(save_directory)
+@withPackage('huggingface_hub')
+def test_from_pretrained(model, tmp_path):
+    save_directory = f'{str(tmp_path / REPO_NAME)}'
+    model.save_pretrained(save_directory)
 
-#     model = model.from_pretrained(save_directory)
-#     assert isinstance(model, DummyModel)
+    model = model.from_pretrained(save_directory)
+    assert isinstance(model, DummyModel)
 
 
 @withPackage('huggingface_hub')

torch_geometric/nn/model_hub.py

@@ -144,10 +144,10 @@ def _from_pretrained(
         revision,
         cache_dir,
         force_download,
-        proxies,
-        resume_download,
         local_files_only,
         token,
+        proxies=None,
+        resume_download=False,
         dataset_name='',
         model_name='',
         map_location='cpu',