update tests

Former-commit-id: 4e92b656e324725048d914946e70867be20032ff
2024-11-02 12:21:41 +08:00
parent 83479c9ef0
commit ba66ac084f
23 changed files with 53 additions and 62 deletions
--- a/tests/data/test_mm_plugin.py
+++ b/tests/data/test_mm_plugin.py
@@ -31,9 +31,9 @@ if TYPE_CHECKING:
    from llamafactory.data.mm_plugin import BasePlugin


-HF_TOKEN = os.environ.get("HF_TOKEN", None)
+HF_TOKEN = os.getenv("HF_TOKEN")

-TINY_LLAMA = os.environ.get("TINY_LLAMA", "llamafactory/tiny-random-Llama-3")
+TINY_LLAMA = os.getenv("TINY_LLAMA", "llamafactory/tiny-random-Llama-3")

 MM_MESSAGES = [
    {"role": "user", "content": "<image>What is in this image?"},