From ded91998c57b8d11ed9119d707c1681ba4c22067 Mon Sep 17 00:00:00 2001
From: rasbt <mail@sebastianraschka.com>
Date: Thu, 30 May 2024 20:59:43 +0000
Subject: [PATCH] update readme tests

---
 tests/test_readme.py | 4 ++--
 tests/test_utils.py  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/test_readme.py b/tests/test_readme.py
index 9d186d3729..0b64d487ea 100644
--- 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
index 63ac0af393..db41baa01f 100644
--- 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",