Fix failing tests (#497)

sayakpaul · danieldk · web-flow · commit 865145b835e1 · 2026-04-21T21:15:53.000+02:00
* feat: add test and docs for get_loaded_kernels.

* fix existing failing tests

* fix 2

* style

---------

Co-authored-by: Daniël de Kok &lt;me@danieldk.eu&gt;
diff --git a/kernels/tests/test_layer.py b/kernels/tests/test_layer.py
@@ -1108,7 +1108,7 @@ def forward(self) -> str:
         }
     ):
         version = kernelize(version, device=device, mode=Mode.INFERENCE)
-        assert version() == "0.2.0"
+        assert version() == 2
 
     with use_kernel_mapping(
         {
diff --git a/kernels/tests/test_loaded_kernels.py b/kernels/tests/test_loaded_kernels.py
@@ -27,7 +27,11 @@ def test_dataclass_shape():
         "module_name",
         "repo_infos",
     )
-    assert tuple(f.name for f in fields(RepoInfos)) == ("repo_id", "revision", "backend")
+    assert tuple(f.name for f in fields(RepoInfos)) == (
+        "repo_id",
+        "revision",
+        "backend",
+    )
 
 
 def test_get_loaded_kernels_returns_copy(fresh_registry):

Original file line number	Diff line number	Diff line change
`@@ -1108,7 +1108,7 @@ def forward(self) -> str:`
`1108`	`1108`	`}`
`1109`	`1109`	`):`
`1110`	`1110`	`version = kernelize(version, device=device, mode=Mode.INFERENCE)`
`1111`		`- assert version() == "0.2.0"`
	`1111`	`+ assert version() == 2`
`1112`	`1112`
`1113`	`1113`	`with use_kernel_mapping(`
`1114`	`1114`	`{`