TST Skips derivative check on 32bit platforms (#17073)

thomasjpfan · web-flow · commit ad6a9f977e37 · 2020-04-29T10:05:37.000+02:00
* TST Checks type for derivative check

* TST Skips test for 32bit linux

* REV Less diffs
diff --git a/sklearn/ensemble/_hist_gradient_boosting/tests/test_loss.py b/sklearn/ensemble/_hist_gradient_boosting/tests/test_loss.py
@@ -9,6 +9,7 @@
 from sklearn.ensemble._hist_gradient_boosting.loss import _LOSSES
 from sklearn.ensemble._hist_gradient_boosting.common import Y_DTYPE
 from sklearn.ensemble._hist_gradient_boosting.common import G_H_DTYPE
+from sklearn.utils._testing import skip_if_32bit
 
 
 def get_derivatives_helper(loss):
@@ -58,8 +59,7 @@ def get_hessians(y_true, raw_predictions):
 ])
 @pytest.mark.skipif(sp_version == (1, 2, 0),
                     reason='bug in scipy 1.2.0, see scipy issue #9608')
-@pytest.mark.skipif(Y_DTYPE != np.float64,
-                    reason='Newton internally uses float64 != Y_DTYPE')
+@skip_if_32bit
 def test_derivatives(loss, x0, y_true):
     # Check that gradients are zero when the loss is minimized on 1D array
     # using Halley's method with the first and second order derivatives
