fix ch07 unit test (#470)

rasbt · Jan 5, 2025 · 2d7ca7e · 2d7ca7e
1 parent 7010908
commit 2d7ca7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ch07/01_main-chapter-code/tests.py b/ch07/01_main-chapter-code/tests.py
@@ -10,7 +10,7 @@
 
 
 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)
     assert result.returncode == 0, f"Script exited with errors: {result.stderr}"