remove onlyrag in txt2kg

Author: xnuohz

test/datasets/test_protein_mpnn_dataset.py

@@ -1,7 +1,8 @@
 from torch_geometric.datasets import ProteinMPNNDataset
-from torch_geometric.testing import onlyOnline, withPackage
+from torch_geometric.testing import onlyLinux, onlyOnline, withPackage
 
 
+@onlyLinux
 @onlyOnline
 @withPackage('pandas')
 def test_protein_mpnn_dataset():

test/llm/models/test_txt2kg.py

@@ -13,7 +13,6 @@
     _multiproc_helper,
     _parse_n_check_triples,
 )
-from torch_geometric.testing import onlyRAG
 
 # ────────────────────────
 # Fixtures
@@ -105,7 +104,6 @@ def fake_llm(txt, **kwargs):
 # ────────────────────────
 
 
-@onlyRAG
 def test_txt2kg_cloud_mode(mock_cloud_llm):
     kg = TXT2KG(NVIDIA_API_KEY="fake-key", local_LM=False)
     kg.add_doc_2_KG("Paris is the capital of France.")
@@ -124,7 +122,6 @@ def test_txt2kg_local_mode(mock_local_llm):
     assert triples == []
 
 
-@onlyRAG
 def test_txt2kg_with_qa_pair(mock_cloud_llm):
     kg = TXT2KG(NVIDIA_API_KEY="fake", local_LM=False)
     qa = ("What is Paris?", "Capital of France")
@@ -134,7 +131,6 @@ def test_txt2kg_with_qa_pair(mock_cloud_llm):
     assert kg.relevant_triples[qa] == []
 
 
-@onlyRAG
 def test_txt2kg_duplicate_qa_warning(capfd, mock_cloud_llm):
     kg = TXT2KG(NVIDIA_API_KEY="fake", local_LM=False)
     qa = ("Q", "A")
@@ -145,7 +141,6 @@ def test_txt2kg_duplicate_qa_warning(capfd, mock_cloud_llm):
     assert "Warning: QA_Pair was already added" in captured.out
 
 
-@onlyRAG
 def test_txt2kg_save_kg(mock_cloud_llm):
     kg = TXT2KG(NVIDIA_API_KEY="fake", local_LM=False)
     kg.add_doc_2_KG("Test document.")

test/llm/utils/test_rag_backend_utils.py

@@ -13,6 +13,7 @@
     preprocess_triplet,
     retrieval_via_pcst,
 )
+from torch_geometric.testing import onlyLinux
 
 
 def test_preprocess_triplet():
@@ -62,6 +63,7 @@ def create_mock_data(num_nodes=3, num_edges=2):
                 edge_idx=edge_idx), textual_nodes, textual_edges
 
 
+@onlyLinux
 def test_empty_graph():
     import pandas as pd
 
@@ -115,6 +117,7 @@ def create_mock_graph_and_triples():
     return triples, graph
 
 
+@onlyLinux
 def test_apply_retrieval_via_pcst_isolated_node():
     triples, graph = create_mock_graph_and_triples()
     model = MockSentenceTransformer()