mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-24 10:08:33 +00:00
b4423b8c4d
*Description of changes:* Speed up GH workflow by installing CPU-only version of torch By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.
46 lines
1.2 KiB
YAML
46 lines
1.2 KiB
YAML
name: CI
|
|
|
|
on: [push, pull_request]
|
|
|
|
jobs:
|
|
type-check:
|
|
strategy:
|
|
max-parallel: 4
|
|
fail-fast: false
|
|
matrix:
|
|
python-version: ["3.11"]
|
|
platform: [ubuntu-latest]
|
|
|
|
runs-on: ${{ matrix.platform }}
|
|
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- name: Set up Python ${{ matrix.python-version }}
|
|
uses: actions/setup-python@v5
|
|
with:
|
|
python-version: ${{ matrix.python-version }}
|
|
- name: Install dependencies
|
|
run: pip install ".[typecheck]" -f https://download.pytorch.org/whl/cpu/torch_stable.html
|
|
- name: Type checks with mypy
|
|
run: mypy src test
|
|
|
|
test:
|
|
strategy:
|
|
max-parallel: 4
|
|
fail-fast: false
|
|
matrix:
|
|
python-version: ["3.11"]
|
|
platform: [ubuntu-latest]
|
|
|
|
runs-on: ${{ matrix.platform }}
|
|
|
|
steps:
|
|
- uses: actions/checkout@v4
|
|
- name: Set up Python ${{ matrix.python-version }}
|
|
uses: actions/setup-python@v5
|
|
with:
|
|
python-version: ${{ matrix.python-version }}
|
|
- name: Install dependencies
|
|
run: pip install ".[test]" -f https://download.pytorch.org/whl/cpu/torch_stable.html
|
|
- name: Test with pytest
|
|
run: pytest
|