Fixed ablation test var names

Future-House · Aug 8, 2023 · f6c69a0 · f6c69a0
1 parent 89ff929
commit f6c69a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_paperqa.py b/tests/test_paperqa.py
@@ -34,7 +34,7 @@ def test_ablations():
                 question="Which page is the statement 'Deep learning (DL) is advancing the boundaries of computational"
                 + "chemistry because it can accurately model non-linear structure-function relationships.' on?"
             ),
-            ablate_summarization=True,
+            disable_summarization=True,
         )
         assert (
             answer.contexts[0].text.text == answer.contexts[0].context
@@ -44,7 +44,7 @@ def test_ablations():
                 question="Which page is the statement 'Deep learning (DL) is advancing the boundaries of computational"
                 + "chemistry because it can accurately model non-linear structure-function relationships.' on?"
             ),
-            ablate_vector_search=True,
+            disable_vector_search=True,
         )