Fix test

Lightning-AI · May 7, 2024 · 4830506 · 4830506
1 parent f9f280a
commit 4830506
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/litgpt/utils.py b/litgpt/utils.py
@@ -444,6 +444,7 @@ def save_hyperparameters(function: callable, checkpoint_dir: Path) -> None:
         ("finetune", "lora"),
         ("finetune", "adapter"),
         ("finetune", "adapter_v2"),
+        ("finetune",),
         ("pretrain",),
     ]
     for known_command in known_commands:

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -267,6 +267,7 @@ def _test_function2(out_dir: Path, foo: bool = False, bar: int = 1):
     "command",
     [
         "any.py",
+        "litgpt finetune",
         "litgpt finetune full",
         "litgpt finetune lora",
         "litgpt finetune adapter",