Fix error in init town (#8)

* Add files via upload * Add files via upload * Add files via upload * Test whether more relaxed mypy could reduce captured error in init-town * sync template * support research_town * fix codespell bug * fix formatting * test pytest * fix mypy errors * try to pass pytest * try to pass pytest * pass all pre-commit * pass all pre-commit * pass all pre-commit * fix isort bug * fix isort bug * fix mypy types * fix ruff * delete install mypy things * fix isort * fix isort * update poetry dependency * add torch support * add isort support * add transformers * fix pytest error * fix pytest error * solve pytest warning * solve pytest --------- Co-authored-by: ft2023 <[email protected]> Co-authored-by: Haofei Yu <[email protected]>
ulab-uiuc · May 9, 2024 · 0ddc19f · 0ddc19f
1 parent c3fd5c3
commit 0ddc19f
Show file tree

Hide file tree

Showing 22 changed files with 1,830 additions and 295 deletions.
diff --git a/.github/workflows/codespell.yml b/.github/workflows/codespell.yml
@@ -0,0 +1,31 @@
+name: codespell
+
+on:
+  # Trigger the workflow on push or pull request,
+  # but only for the main branch
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  codespell:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10"]
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install codespell==2.2.6 tomli==2.0.1
+    - name: Spelling check with codespell
+      run: |
+        codespell --toml pyproject.toml
diff --git a/.github/workflows/isort.yml b/.github/workflows/isort.yml
@@ -0,0 +1,31 @@
+name: isort
+
+on:
+  # Trigger the workflow on push or pull request,
+  # but only for the main branch
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  isort:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10"]
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install isort==5.13.2
+    - name: Run isort
+      run: |
+        isort . --check-only
diff --git a/.github/workflows/mypy.yml b/.github/workflows/mypy.yml
@@ -1,18 +1,28 @@
 name: Mypy
-on: [push]
+on:
+  # Trigger the workflow on push or pull request,
+  # but only for the main branch
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
 
 jobs:
   Static-Type-Checking:
     runs-on: ubuntu-latest
     strategy:
       max-parallel: 5
+      matrix:
+        python-version: ["3.9", "3.10"]
 
     steps:
-    - uses: actions/checkout@v3
-    - name: Set up Python 3.11
-      uses: actions/setup-python@v4
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
       with:
-        python-version: 3.11.2
+        python-version: ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
         curl -sSL https://install.python-poetry.org | python3
@@ -21,4 +31,4 @@ jobs:
       run: |
         # Run this mypy instance against our main package.
         poetry run pip install types-protobuf==4.24.0.4
-        poetry run mypy --strict .
+        poetry run mypy --config-file pyproject.toml .
diff --git a/.github/workflows/pre-commit.yml b/.github/workflows/pre-commit.yml
@@ -8,9 +8,13 @@ on:
 jobs:
   pre-commit:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10"]
+
     steps:
-    - uses: actions/checkout@v3
-    - name: Set up Python 3.11
-      uses: actions/setup-python@v4
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
       with:
-        python-version: 3.11.2
+        python-version: ${{ matrix.python-version }}
diff --git a/.github/workflows/ruff.yml b/.github/workflows/ruff.yml
@@ -0,0 +1,31 @@
+name: ruff
+
+on:
+  # Trigger the workflow on push or pull request,
+  # but only for the main branch
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  ruff:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.9", "3.10"]
+    steps:
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
+      with:
+        python-version: ${{ matrix.python-version }}
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install ruff==0.1.5
+    - name: Analysing the code with ruff
+      run: |
+        ruff check .
diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
@@ -1,18 +1,23 @@
 name: Pytest
-on: [push]
+on:
+  pull_request:
+  push:
+    branches: [main]
 
 jobs:
   Pytest:
     runs-on: ubuntu-latest
     strategy:
       max-parallel: 5
+      matrix:
+        python-version: ["3.9", "3.10"]
 
     steps:
-    - uses: actions/checkout@v3
-    - name: Set up Python 3.11
-      uses: actions/setup-python@v4
+    - uses: actions/checkout@v2
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v2
       with:
-        python-version: 3.11.2
+        python-version: ${{ matrix.python-version }}
     - name: Install dependencies
       run: |
         curl -sSL https://install.python-poetry.org | python3

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1,17 +1,4 @@
 repos:
-- repo: local
-  hooks:
-    - id: poetry-update
-      name: Poetry Update
-      entry: poetry update
-      language: system
-      # Optional: specify types or files to trigger this hook
-      # files: pyproject.toml
-      # Always run this hook (not recommended for update; shown for illustrative purposes):
-      always_run: true
-      # Optional: prevent commit if the lock file changes
-      stages: [pre-commit]
-      pass_filenames: false
 -   repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v3.2.0
     hooks:
@@ -24,16 +11,17 @@ repos:
     hooks:
     -   id: prettier
         types_or: [html]
--   repo: https://github.com/psf/black
-    rev: 22.12.0
-    hooks:
-    -   id: black
-        args: [--line-length=79]
--   repo: https://github.com/pycqa/isort
-    rev: 5.12.0
-    hooks:
-    -   id: isort
-        args: ["--profile", "black", --line-length=72]
+- repo: https://github.com/astral-sh/ruff-pre-commit
+  rev: v0.3.5  # Ruff version
+  hooks:
+    - id: ruff
+      types_or: [python, pyi, jupyter]
+      args: [--fix]
+
+- repo: https://github.com/pre-commit/mirrors-isort
+  rev: v5.10.1  # Use the latest isort version
+  hooks:
+    - id: isort  # This will sort imports automatically
 -   repo: https://github.com/kynan/nbstripout
     rev: 0.6.0
     hooks:

diff --git a/README.md b/README.md
@@ -1 +1 @@
-# research-town
+# research-town