Skip to content

Commit

Permalink
[CI] Test train workflow on GPU machine (#590)
Browse files Browse the repository at this point in the history
* add test workflow on GPU for train

* fix conda path

* fix conflicting workdir

* only run on non-draft PRs

* run also on refactoring branch
  • Loading branch information
NicolasGensollen authored May 23, 2024
1 parent 2861e9d commit f5de251
Showing 1 changed file with 53 additions and 0 deletions.
53 changes: 53 additions & 0 deletions .github/workflows/test_train.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
name: Train Tests (GPU)

on:
push:
branches: ["dev", "refactoring"]
pull_request:
branches: ["dev", "refactoring"]

permissions:
contents: read

concurrency:
group: '${{ github.workflow }} @ ${{ github.event.pull_request.head.label || github.head_ref || github.ref }}'
cancel-in-progress: true

env:
POETRY_VERSION: '1.8.3'
PYTHON_VERSION: '3.11'

jobs:
test-train-gpu:
if: github.event.pull_request.draft == false
runs-on:
- self-hosted
- Linux
- ubuntu
- gpu
steps:
- uses: actions/checkout@v4
- uses: snok/install-poetry@v1
with:
version: ${{ env.POETRY_VERSION }}
virtualenvs-create: false
- uses: actions/setup-python@v5
with:
python-version: ${{ env.PYTHON_VERSION }}
- name: Run tests for Train on GPU
run: |
make env.conda
source "${HOME}/miniconda3/etc/profile.d/conda.sh"
conda activate "${{ github.workspace }}"/env
make install
cd tests
poetry run pytest --verbose \
--junitxml=./test-reports/test_train_report.xml \
--disable-warnings \
--verbose \
--basetemp=$HOME/actions_runner_workdir/train \
--input_data_directory=/mnt/data/clinicadl_data_ci/data_ci \
-k test_train
- name: Cleaning
run: |
rm -rf $HOME/actions_runner_workdir/train/*

0 comments on commit f5de251

Please sign in to comment.