mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-08-16 20:51:51 +00:00
fix ch07 unit test (#470)
This commit is contained in:
parent
701090815e
commit
2d7ca7ee4b
@ -10,7 +10,7 @@ import subprocess
|
|||||||
|
|
||||||
|
|
||||||
def test_gpt_class_finetune():
|
def test_gpt_class_finetune():
|
||||||
command = ["python", "ch06/01_main-chapter-code/gpt_class_finetune.py", "--test_mode"]
|
command = ["python", "ch07/01_main-chapter-code/gpt_instruction_finetuning.py", "--test_mode"]
|
||||||
|
|
||||||
result = subprocess.run(command, capture_output=True, text=True)
|
result = subprocess.run(command, capture_output=True, text=True)
|
||||||
assert result.returncode == 0, f"Script exited with errors: {result.stderr}"
|
assert result.returncode == 0, f"Script exited with errors: {result.stderr}"
|
||||||
|
Loading…
x
Reference in New Issue
Block a user