Fixed the load artifact test (#128)

AlignmentResearch · Mar 12, 2024 · 36c41e0 · 36c41e0
1 parent 65f3135
commit 36c41e0
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_load_artifact.py b/tests/test_load_artifact.py
@@ -15,9 +15,9 @@ def test_load_lens_artifact_raises_smoke():
 def test_list_available_lens_artifacts_smoke():
     artifacts = available_lens_artifacts("AlignmentResearch/tuned-lens", "space")
     assert len(artifacts) > 0
-    artifacts = available_lens_artifacts(
-        "AlignmentResearch/tuned-lens",
-        "space",
-        "revision_does_not_exist",
-    )
-    assert len(artifacts) == 0
+    with pytest.raises(FileNotFoundError):
+        artifacts = available_lens_artifacts(
+            "AlignmentResearch/tuned-lens",
+            "space",
+            "revision_does_not_exist",
+        )