olmoearth_pretrain/.github/workflows/tests.yaml at a95b7c73a209da4f745ec83f245ac79c78561311 · allenai/olmoearth_pretrain · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
name: Tests

on:
  push:
    branches:
      - main
  pull_request:
    branches:
      - main

jobs:
  test:
    name: Test (full dependencies)
    runs-on: ubuntu-latest

    steps:
    - uses: actions/checkout@v4

    - name: Set up Python
      uses: actions/setup-python@v5
      with:
        python-version-file: "pyproject.toml"

    - name: Install uv
      uses: astral-sh/setup-uv@v6
      with:
        version: "0.9.5"

    - name: Install dependencies
      run: |
        # Install all extras (includes training with olmo-core, eval, dev, etc.)
        uv sync --locked --all-extras

    - name: Run unit/integration tests
      run: |
        uv run pytest -n auto tests/

    - name: Run model loader tests (with olmo-core)
      run: |
        uv run pytest -v tests_minimal_deps/

  test-inference-only:
    name: Test (inference-only dependencies)
    runs-on: ubuntu-latest

    steps:
    - uses: actions/checkout@v4

    - name: Set up Python
      uses: actions/setup-python@v5
      with:
        python-version-file: "pyproject.toml"

    - name: Install uv
      uses: astral-sh/setup-uv@v6
      with:
        version: "0.9.5"

    - name: Run model loader tests (without olmo-core)
      run: |
        # Run with base deps + dev extra (no training = no olmo-core)
        uv run --extra dev pytest -v tests_minimal_deps/