Add CI tests for chapter 7 (#239)

This commit is contained in:
Sebastian Raschka
2024-06-22 08:57:18 -05:00
committed by GitHub
parent 0114dee9f6
commit eb85c43bc3
3 changed files with 81 additions and 23 deletions

View File

@@ -0,0 +1,16 @@
# Copyright (c) Sebastian Raschka under Apache License 2.0 (see LICENSE.txt).
# Source for "Build a Large Language Model From Scratch"
# - https://www.manning.com/books/build-a-large-language-model-from-scratch
# Code: https://github.com/rasbt/LLMs-from-scratch
# File for internal use (unit tests)
import subprocess
def test_gpt_class_finetune():
command = ["python", "ch06/01_main-chapter-code/gpt_class_finetune.py", "--test_mode"]
result = subprocess.run(command, capture_output=True, text=True)
assert result.returncode == 0, f"Script exited with errors: {result.stderr}"