update readme tests

Lightning-AI · May 30, 2024 · ded9199 · ded9199
1 parent c03ff3f
commit ded9199
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_readme.py b/tests/test_readme.py
@@ -107,7 +107,7 @@ def test_pretrain_model():
     OUT_DIR = Path("out") / "custom_pretrained"
     pretrain_command = [
         "litgpt", "pretrain",
-        "--model_name", "pythia-14m",
+        "pythia-14m",
         "--tokenizer_dir", str("checkpoints" / REPO_ID),
         "--data", "TextFiles",
         "--data.train_data_path", str(CUSTOM_TEXTS_DIR),
@@ -127,7 +127,7 @@ def test_continue_pretrain_model():
     OUT_DIR = Path("out") / "custom_continue_pretrained"
     pretrain_command = [
         "litgpt", "pretrain",
-        "--model_name", "pythia-14m",
+        "pythia-14m",
         "--initial_checkpoint", str("checkpoints" / REPO_ID),
         "--tokenizer_dir", str("checkpoints" / REPO_ID),
         "--data", "TextFiles",

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -270,7 +270,7 @@ def _test_function2(out_dir: Path, foo: bool = False, bar: int = 1):
 @pytest.mark.parametrize(
     "command",
     [
-        "any.py",
+        #"any.py",
         "litgpt finetune",
         "litgpt finetune_full",
         "litgpt finetune_lora",