fix: ci (#1)

* fix: ci

Author: shenxiangzhuang

.github/workflows/build_docs.yaml

@@ -10,9 +10,11 @@ jobs:
   deploy:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - name: Install Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
+        with:
+          python-version: '3.11'
       - name: Setup pandoc
         run: |
           wget https://github.com/jgm/pandoc/releases/download/2.18/pandoc-2.18-1-amd64.deb && sudo dpkg -i pandoc-2.18-1-amd64.deb
@@ -29,7 +31,7 @@ jobs:
           path: ./.venv
           key: venv-${{ hashFiles('poetry.lock') }}
       - name: Install the project dependencies
-        run: poetry install
+        run: poetry install --only dev
 
       - name: Build docs
         run: poetry run mkdocs gh-deploy --force

.github/workflows/python-publish.yml

@@ -21,11 +21,11 @@ jobs:
     environment: publish
 
     steps:
-    - uses: actions/checkout@v3
-    - name: Set up Python
-      uses: actions/setup-python@v3
+    - uses: actions/checkout@v4
+    - name: Install Python
+      uses: actions/setup-python@v5
       with:
-        python-version: '3.x'
+        python-version: '3.11'
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip

.github/workflows/test.yaml

@@ -11,15 +11,18 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: [ "3.11"]
+        python-version: ["3.11"]
         poetry-version: ["1.7"]
 #        os: [ubuntu-latest, macos-latest, windows-latest]
         os: [ubuntu-latest]
     runs-on: ${{ matrix.os }}
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
       - name: Install Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+
         # see details (matrix, python-version, python-version-file, etc.)
         # https://github.com/actions/setup-python
       - name: Install poetry
@@ -37,13 +40,14 @@ jobs:
         run: poetry install
 
       - name: Run tests and generate coverage report
-        run: poetry run pytest --doctest-modules -v --cov=mppt --cov-fail-under 0 --cov-report=term --cov-report=xml --cov-report=html
+        run: poetry run pytest --doctest-modules -v --cov=toyllm --cov-fail-under 0 --cov-report=term --cov-report=xml --cov-report=html
 
       - name: Upload coverage to Codecov
-        uses: codecov/codecov-action@v3
-        if: success() && matrix.os == 'ubuntu-latest' && matrix.python-version == '3.8'
+        uses: codecov/codecov-action@v4.0.1
+        if: success() && matrix.os == 'ubuntu-latest' && matrix.python-version == '3.11'
         with:
           token: ${{ secrets.CODECOV_TOKEN }}
+          slug: toy-ai/toyllm
           files: ./coverage.xml
           fail_ci_if_error: true
           verbose: true

CHANGELOG.md

@@ -2,4 +2,3 @@
 
 
 ## [Unreleased]
-

README.md

@@ -2,4 +2,3 @@
 
 ToyLLM is a simple language model that can be used to generate text.
 It is based on the [GPT-2](https://huggingface.co/transformers/model_doc/gpt2.html) model.
-

tests/test_device.py

@@ -0,0 +1,6 @@
+from toyllm.device import get_device
+
+
+def test_device():
+    device = get_device()
+    assert device.type in ("cuda", "cpu")

toyllm/device/__init__.py

@@ -0,0 +1,4 @@
+import torch
+
+def get_device() -> torch.device:
+    return torch.device("cuda" if torch.cuda.is_available() else "cpu")