From 4ef261743a10336ed6eaf9493bb6496283912d27 Mon Sep 17 00:00:00 2001 From: Thomas Viehmann Date: Fri, 13 Sep 2024 11:32:55 +0200 Subject: [PATCH] try to find segfaulting test --- tests/test_api.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_api.py b/tests/test_api.py index 11de6b28fc..f00ec237c2 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -74,13 +74,14 @@ def test_generate_token_ids(mock_llm): assert output_ids.shape[0] > len(prompt) +@pytest.mark.skipif(sys.platform == "darwin", reason="segfaults on macos-14") def test_calculate_number_of_devices(): assert calculate_number_of_devices(1) == 1 assert calculate_number_of_devices([0, 1, 2]) == 3 assert calculate_number_of_devices(None) == 0 -@pytest.mark.skipif(sys.platform == "darwin", reason="segfaults on macos-14") +#@pytest.mark.skipif(sys.platform == "darwin", reason="segfaults on macos-14") def test_llm_load_random_init(tmp_path): download_from_hub(repo_id="EleutherAI/pythia-14m", tokenizer_only=True, checkpoint_dir=tmp_path)