MilesCranmer commited on
Commit
4655d23
1 Parent(s): 1122900

Skip jax tests on old python CI

Browse files
Files changed (1) hide show
  1. .github/workflows/CI.yml +6 -0
.github/workflows/CI.yml CHANGED
@@ -32,10 +32,12 @@ jobs:
32
  julia-version: ['1']
33
  python-version: ['3.11']
34
  os: [ubuntu-latest]
 
35
  include:
36
  - julia-version: ['1.6']
37
  python-version: ['3.7']
38
  os: [ubuntu-latest]
 
39
 
40
  steps:
41
  - uses: actions/checkout@v3
@@ -67,12 +69,16 @@ jobs:
67
  coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test cli
68
  - name: "Install JAX"
69
  run: pip install jax jaxlib # (optional import)
 
70
  - name: "Run JAX tests"
71
  run: coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test jax
 
72
  - name: "Install Torch"
73
  run: pip install torch # (optional import)
 
74
  - name: "Run Torch tests"
75
  run: coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test torch
 
76
  - name: "Run custom env tests"
77
  run: coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test env
78
  - name: "Coveralls"
 
32
  julia-version: ['1']
33
  python-version: ['3.11']
34
  os: [ubuntu-latest]
35
+ test-id: ['main']
36
  include:
37
  - julia-version: ['1.6']
38
  python-version: ['3.7']
39
  os: [ubuntu-latest]
40
+ test-id: ['include']
41
 
42
  steps:
43
  - uses: actions/checkout@v3
 
69
  coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test cli
70
  - name: "Install JAX"
71
  run: pip install jax jaxlib # (optional import)
72
+ if: ${{ matrix.test-id == 'main' }}
73
  - name: "Run JAX tests"
74
  run: coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test jax
75
+ if: ${{ matrix.test-id == 'main' }}
76
  - name: "Install Torch"
77
  run: pip install torch # (optional import)
78
+ if: ${{ matrix.test-id == 'main' }}
79
  - name: "Run Torch tests"
80
  run: coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test torch
81
+ if: ${{ matrix.test-id == 'main' }}
82
  - name: "Run custom env tests"
83
  run: coverage run --append --source=pysr --omit='*/test/*,*/feynman_problems.py' -m pysr.test env
84
  - name: "Coveralls"