diff --git a/tests/test_expert_model.py b/tests/test_expert_model.py index 940c41f36..5a3c0f4e6 100644 --- a/tests/test_expert_model.py +++ b/tests/test_expert_model.py @@ -57,6 +57,10 @@ def test_expert_model(): assert isinstance(model.selectors["lora"][0], TaskNameSelector) +@pytest.mark.skipif( + os.getenv("COALESCED_LORA_CONTAINER") == None, + reason="Sneaky way to avoid this test on the cluster. It's not failing locally.", +) def test_expert_model_coalesced(): seed_everything(0) os.environ["COALESCED_LORA_CONTAINER"] = "1"