Fix skip test

tests/distributed/test_model_parallelization.py

@@ -526,9 +526,6 @@ def test_llama_v2_gqa(
         if num_kv_heads < tp_size and (not from_pretrained):
             pytest.skip("This case will  not work here because we set the seed. We can skip.")
 
-        if not from_pretrained and not lazy_load and not sequence_parallel_enabled:
-            pytest.skip("It seems to be a compiler bug, to investigate")
-
         model_name_or_path = Path(tmpdir) / "llama_v2_gqa"
 
         # Since we are creating the model from config, we actually first create a model locally from config and then

tests/test_cache_utils.py

@@ -484,7 +484,7 @@ def test_neuron_hash_is_private(self):
 
 @is_trainium_test
 @is_staging_test
-@pytest.skip("This is not needed anymore and will be removed.")
+@pytest.mark.skip("This is not needed anymore and will be removed.")
 class CachedModelOnTheHubTestCase(StagingTestMixin, TestCase):
     def test_push_to_hub_fails_with_private_model_and_public_repo(self):
         with TemporaryDirectory() as tmpdirname: