try to find segfaulting test

Lightning-AI · Sep 13, 2024 · 4ef2617 · 4ef2617
1 parent 671ba25
commit 4ef2617
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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)