Update unittest

rmazzine · Aug 31, 2023 · 19ce4e4 · 19ce4e4
1 parent 00077cc
commit 19ce4e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_cf_finder.py b/tests/test_cf_finder.py
@@ -456,7 +456,7 @@ def test_find_tabular_example(
         self.assertEqual(mock_fine_tuning.call_args[1]['verbose'], verbose)
 
         # Check call for _CFTabular
-        self.assertEqual(len(mock_CFTabular.call_args[1]), 7)
+        self.assertEqual(len(mock_CFTabular.call_args[1]), 8)
         self.assertListEqual(mock_CFTabular.call_args[1]['factual'].tolist(), factual.tolist())
         self.assertListEqual(mock_CFTabular.call_args[1]['factual_vector'].tolist(), factual.tolist())
         self.assertEqual(mock_CFTabular.call_args[1]['cf_vectors'], mock_fine_tuning().__getitem__())