cwa.yaml

name: cwa

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

jobs:

  prepare-artifacts:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - name: Setup Python
        uses: actions/setup-python@v5
        with:
          python-version: "3.8"
      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          python -m pip install -e .  # install as editable to locate the torch hub cache more easily

      - name: Download test files
        run: |
          wget https://wearables-files.ndph.ox.ac.uk/files/data/samples/mini-sample.csv.gz --quiet
          wget https://wearables-files.ndph.ox.ac.uk/files/data/samples/ax3/tiny-sample.cwa.gz --quiet

      # Perform a run to cache the torch hub repo and download the model files (SSL)
      - name: Smoke test
        run: stepcount mini-sample.csv.gz

      # Perform a run to download the model files (RF)
      - name: Smoke test (RF)
        run: stepcount mini-sample.csv.gz -t rf

      - name: Upload test files
        uses: actions/upload-artifact@v4
        with:
          name: test-files
          path: |
            mini-sample.csv.gz
            tiny-sample.cwa.gz

      - name: Upload model joblib files
        uses: actions/upload-artifact@v4
        with:
          name: model-files
          path: src/stepcount/*.joblib.lzma  # our model files have the extension .joblib.lzma

      - name: Upload torch_hub_cache
        uses: actions/upload-artifact@v4
        with:
          name: torch_hub_cache
          path: src/stepcount/torch_hub_cache

  cwa:
    needs: prepare-artifacts
    runs-on: ${{matrix.os}}
    strategy:
      matrix:
        os: [ubuntu-latest, macos-latest]
        python-version: ["3.8", "3.9", "3.10"]

    steps:
      - uses: actions/checkout@v4

      - name: Setup Python ${{ matrix.python-version }}
        uses: actions/setup-python@v5
        with:
          python-version: ${{ matrix.python-version }}

      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          python -m pip install -e .  # install as editable to locate the torch hub cache more easily

      - name: Download test files
        uses: actions/download-artifact@v4
        with:
          name: test-files

      - name: Download model files
        uses: actions/download-artifact@v4
        with:
          name: model-files
          path: src/stepcount

      - name: Download torch_hub_cache
        uses: actions/download-artifact@v4
        with:
          name: torch_hub_cache
          path: src/stepcount/torch_hub_cache

      - name: Smoke test, mini-sample.csv.gz
        run: stepcount mini-sample.csv.gz

      - name: Smoke test, tiny-sample.cwa.gz
        run: stepcount tiny-sample.cwa.gz

      - name: Smoke test, mini-sample.csv.gz (RF)
        run: stepcount mini-sample.csv.gz -t rf

      - name: Smoke test, tiny-sample.cwa.gz (RF)
        run: stepcount tiny-sample.cwa.gz -t rf