fix: avoid non version locking tests

Author: drbh

kernels/tests/test_basic.py

@@ -3,7 +3,7 @@
 import pytest
 import torch
 import torch.nn.functional as F
-from huggingface_hub.errors import RepositoryNotFoundError
+from huggingface_hub.errors import HfHubHTTPError
 
 from kernels import get_kernel, get_local_kernel, has_kernel, install_kernel
 
@@ -131,6 +131,7 @@ def test_has_kernel(kernel_exists):
     assert has_kernel(repo_id, revision=revision) == kernel
 
 
+@pytest.mark.skip(reason="Tags are not supported on kernel repos")
 def test_version_old():
     # Remove once we drop support for version specs.
     kernel = get_kernel("kernels-test/versions")
@@ -243,7 +244,7 @@ def test_local_overrides(monkeypatch, local_kernel_path):
 
     # Ensure that we are testing with a non-existing kernel, so that we know
     # that the kernel must be local.
-    with pytest.raises(RepositoryNotFoundError):
+    with pytest.raises(HfHubHTTPError):
         get_kernel(f"kernels-test/{package_name}")
 
     with monkeypatch.context() as m:

kernels/tests/test_kernel_locking.py

@@ -75,6 +75,7 @@ def forward(self) -> str:
         assert version() == "1"
 
 
+@pytest.mark.skip(reason="Tags are not supported on kernel repos")
 def test_layer_locked_old(device):
     project_dir = Path(__file__).parent / "layer_locking_old"
 
@@ -154,6 +155,7 @@ def forward(self) -> str:
     assert version() == "0.0.0"
 
 
+@pytest.mark.skip(reason="Tags are not supported on kernel repos")
 def test_func_locked_old(device):
     project_dir = Path(__file__).parent / "layer_locking_old"
 

kernels/tests/test_layer.py

@@ -4,7 +4,7 @@
 import pytest
 import torch
 import torch.nn as nn
-from huggingface_hub.errors import RepositoryNotFoundError
+from huggingface_hub.errors import HfHubHTTPError
 from torch.nn import functional as F
 
 from kernels import (
@@ -1128,6 +1128,7 @@ def test_kernel_modes_cross_fallback():
         assert linear.n_calls == 2
 
 
+@pytest.mark.skip(reason="Tags are not supported on kernel repos")
 def test_layer_versions_old(device):
     @use_kernel_forward_from_hub("Version")
     class Version(nn.Module):
@@ -1319,7 +1320,7 @@ def test_local_overrides_layer(monkeypatch, local_kernel_path):
     model = SiluAndMulWithKernel()
 
     with use_kernel_mapping(mapping, inherit_mapping=False):
-        with pytest.raises(RepositoryNotFoundError):
+        with pytest.raises(HfHubHTTPError):
             kernelize(model, device="cuda", mode=Mode.INFERENCE)
 
     with monkeypatch.context() as m: