diff --git a/tests/test_runs/test_run_functions.py b/tests/test_runs/test_run_functions.py index 89b6ef0e6..a9abcd05e 100644 --- a/tests/test_runs/test_run_functions.py +++ b/tests/test_runs/test_run_functions.py @@ -183,7 +183,11 @@ def _rerun_model_and_compare_predictions(self, run_id, model_prime, seed, create predictions_prime = run_prime._generate_arff_dict() self._compare_predictions(predictions, predictions_prime) - pd.testing.assert_frame_equal(run.predictions, run_prime.predictions) + pd.testing.assert_frame_equal( + run.predictions, + run_prime.predictions, + check_dtype=False, # Loaded ARFF reads NUMERIC as float, even if integer. + ) def _perform_run( self,