ADD: report computation in tests

artefactory · Oct 21, 2024 · 2b3fac5 · 2b3fac5
1 parent f47fc88
commit 2b3fac5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration_tests/models/test_conditional_logit.py b/tests/integration_tests/models/test_conditional_logit.py
@@ -55,7 +55,7 @@ def test_mode_canada_fit():
     canada_dataset = load_modecanada(as_frame=False, preprocessing="tutorial")
 
     model = ConditionalLogit(coefficients=coefficients)
-    model.fit(canada_dataset)
+    model.fit(canada_dataset, get_report=True)
 
     total_nll = model.evaluate(canada_dataset) * len(canada_dataset)
     assert total_nll <= 1874.4, f"Got NLL: {total_nll}"