Skip to content

Commit 3cb8198

Browse files
committed
Add torch_script test to CI
1 parent d4d4cf6 commit 3cb8198

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

.github/workflows/tests.yml

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -105,6 +105,21 @@ jobs:
105105
- name: Test with PyTest
106106
run: uv run pytest -v -rsx -n 2 -m "torch_export"
107107

108+
test_torch_script:
109+
runs-on: ubuntu-latest
110+
steps:
111+
- uses: actions/checkout@v4
112+
- name: Set up Python
113+
uses: astral-sh/setup-uv@v5
114+
with:
115+
python-version: "3.10"
116+
- name: Install dependencies
117+
run: uv pip install -r requirements/required.txt -r requirements/test.txt
118+
- name: Show installed packages
119+
run: uv pip list
120+
- name: Test with PyTest
121+
run: uv run pytest -v -rsx -n 2 -m "torch_script"
122+
108123
minimum:
109124
runs-on: ubuntu-latest
110125
steps:

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ markers = [
6666
"logits_match",
6767
"compile",
6868
"torch_export",
69+
"torch_script",
6970
]
7071

7172
[tool.coverage.run]

0 commit comments

Comments
 (0)