Update test_neural_reasoner_accuracy.py

dice-group · Dec 5, 2024 · 0140072 · 0140072
1 parent 19f8bd4
commit 0140072
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_neural_reasoner_accuracy.py b/tests/test_neural_reasoner_accuracy.py
@@ -33,9 +33,9 @@ def test_retrieval_performance():
 
     # (2) Initialize Neural OWL Reasoner.
     if path_kge_model:
-        neural_owl_reasoner = OWLNeuralReasoner(path_neural_embedding=path_kge_model, gamma=gamma)
+        neural_owl_reasoner = OWLNeuralReasoner(path_neural_embedding=path_kge_model, gamma=gamma, max_cache_size=0)
     else:
-        neural_owl_reasoner = OWLNeuralReasoner(path_of_kb=path_kg, gamma=gamma)
+        neural_owl_reasoner = OWLNeuralReasoner(path_of_kb=path_kg, gamma=gamma, max_cache_size=0)
 
     # Fix the random seed.
     random.seed(seed)