diff --git a/tests/test_lgbm_compression.py b/tests/test_lgbm_compression.py index eaaf5ec..64037e0 100644 --- a/tests/test_lgbm_compression.py +++ b/tests/test_lgbm_compression.py @@ -53,6 +53,7 @@ def test_compressed_size(diabetes_toy_df, lgbm_regressor, tmp_path): assert size_compressed < 0.6 * size +@pytest.mark.xfail(reason="Flaky test") @pytest.mark.parametrize("compression_method", ["no", "lzma", "gzip", "bz2"]) def test_dump_times(diabetes_toy_df, lgbm_regressor, tmp_path, compression_method): lgbm_regressor.fit(*diabetes_toy_df) @@ -64,6 +65,7 @@ def test_dump_times(diabetes_toy_df, lgbm_regressor, tmp_path, compression_metho assert dump_time_compressed < factor * dump_time_uncompressed +@pytest.mark.xfail(reason="Flaky test") @pytest.mark.parametrize("compression_method", ["no", "lzma", "gzip", "bz2"]) def test_load_times(diabetes_toy_df, lgbm_regressor, tmp_path, compression_method): lgbm_regressor.fit(*diabetes_toy_df) diff --git a/tests/test_sklearn_compression.py b/tests/test_sklearn_compression.py index a2fcd2b..0cb649c 100644 --- a/tests/test_sklearn_compression.py +++ b/tests/test_sklearn_compression.py @@ -99,6 +99,7 @@ def test_compression_size(diabetes_toy_df, random_forest_regressor, tmp_path): assert size_dtype_reduction < 0.5 * size_no_reduction +@pytest.mark.xfail(reason="Flaky test") @pytest.mark.parametrize("compression_method", ["no", "lzma", "gzip", "bz2"]) def test_dump_times( diabetes_toy_df, random_forest_regressor, tmp_path, compression_method @@ -113,6 +114,7 @@ def test_dump_times( assert dump_time_compressed < factor * dump_time_uncompressed +@pytest.mark.xfail(reason="Flaky test") @pytest.mark.parametrize("compression_method", ["no", "lzma", "gzip", "bz2"]) def test_load_times( diabetes_toy_df, random_forest_regressor, tmp_path, compression_method