added assert for pytorch model file

Lightning-AI · Dec 15, 2024 · 1122880 · 1122880
1 parent 6e6329e
commit 1122880
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_readme.py b/tests/test_readme.py
@@ -40,6 +40,7 @@ def test_download_model():
     s = Path("checkpoints") / repo_id
     assert f"Saving converted checkpoint to {str(s)}" in output
     assert ("checkpoints" / REPO_ID).exists()
+    assert ("checkpoints" / REPO_ID / "pytorch_model.bin").exists()
 
     # Also test valid but unsupported repo IDs
     command = ["litgpt", "download", "CohereForAI/aya-23-8B"]