Skip to content

Commit f03f76e

Browse files
committed
Fix magic mock for python < 3.9
1 parent 053b6c2 commit f03f76e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

test/test_automl/test_automl.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1053,7 +1053,9 @@ def test_fit_performs_dataset_compression_without_precision_with_int(
10531053
auto.fit(X, y, only_return_configuration_space=True)
10541054

10551055
assert mock_reduce_dataset.call_count == 1
1056-
assert mock_reduce_dataset.call_args.call_args["operations"] == ["subsample"]
1056+
1057+
args, kwargs = mock_reduce_dataset.call_args
1058+
assert kwargs["operations"] == ["subsample"]
10571059

10581060

10591061
@pytest.mark.parametrize("dataset_compression", [True])

0 commit comments

Comments
 (0)