yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-08-29 19:10:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
rasbt 2024-05-12 19:03:14 -05:00
parent 37c33d6fee
commit c8bcdf5206
4 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/basic-tests-linux.yml vendored
View File

@ -41,7 +41,7 @@ jobs:
pytest setup/02_installing-python-libraries/tests.py pytest setup/02_installing-python-libraries/tests.py
pytest ch04/01_main-chapter-code/tests.py pytest ch04/01_main-chapter-code/tests.py
pytest ch05/01_main-chapter-code/tests.py pytest ch05/01_main-chapter-code/tests.py
pytest ch06/01_main-chapter-code/gpt-class-finetune.py --test_mode pytest ch06/01_main-chapter-code/tests.py
- name: Validate Selected Jupyter Notebooks - name: Validate Selected Jupyter Notebooks
run: | run: |

2
.github/workflows/basic-tests-macos.yml vendored
View File

@ -41,7 +41,7 @@ jobs:
pytest setup/02_installing-python-libraries/tests.py pytest setup/02_installing-python-libraries/tests.py
pytest ch04/01_main-chapter-code/tests.py pytest ch04/01_main-chapter-code/tests.py
pytest ch05/01_main-chapter-code/tests.py pytest ch05/01_main-chapter-code/tests.py
pytest ch06/01_main-chapter-code/gpt-class-finetune.py --test_mode pytest ch06/01_main-chapter-code/tests.py
- name: Validate Selected Jupyter Notebooks - name: Validate Selected Jupyter Notebooks
run: | run: |

2
.github/workflows/basic-tests-windows.yml vendored
View File

@ -44,7 +44,7 @@ jobs:
pytest setup/02_installing-python-libraries/tests.py pytest setup/02_installing-python-libraries/tests.py
pytest ch04/01_main-chapter-code/tests.py pytest ch04/01_main-chapter-code/tests.py
pytest ch05/01_main-chapter-code/tests.py pytest ch05/01_main-chapter-code/tests.py
pytest ch06/01_main-chapter-code/gpt-class-finetune.py --test_mode pytest ch06/01_main-chapter-code/tests.py
- name: Validate Selected Jupyter Notebooks - name: Validate Selected Jupyter Notebooks
shell: bash shell: bash

18
ch06/01_main-chapter-code/tests.py Normal file
View File

@ -0,0 +1,18 @@
# 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
import pytest
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}"