Skip to content
.github/workflows/convergence-tests.yml

indent #127

Sign in to view logs

indent

indent #127

Workflow file for this run

.github/workflows/convergence-tests.yml at d242e3a
name: "BioNeMo Model Convergence Tests"

Check failure on line 1 in .github/workflows/convergence-tests.yml

View workflow run for this annotation

GitHub Actions / .github/workflows/convergence-tests.yml

Invalid workflow file 

(Line: 37, Col: 23): Unrecognized function: 'split'. Located at position 195 within expression: github.event_name == 'schedule' && fromJSON('["esm2_native_te_650m", "esm2_native_te_3b", "esm2_native_te_15b"]') ||
  github.event.inputs.model_configs != '' && fromJSON(format('["{0}"]', join(split(github.event.inputs.model_configs, ','), '","'))) ||
  fromJSON('["esm2_native_te_650m", "esm2_native_te_3b", "esm2_native_te_15b"]'), (Line: 37, Col: 23): Unexpected value '${{
  github.event_name == 'schedule' && fromJSON('["esm2_native_te_650m", "esm2_native_te_3b", "esm2_native_te_15b"]') ||
  github.event.inputs.model_configs != '' && fromJSON(format('["{0}"]', join(split(github.event.inputs.model_configs, ','), '","'))) ||
  fromJSON('["esm2_native_te_650m", "esm2_native_te_3b", "esm2_native_te_15b"]')
}}'
on:
workflow_dispatch:
inputs:
gpu_type:
description: "GPU type to use"
required: true
default: "h100-sxm"
type: choice
options:
- h200
- h100-sxm
- a100-80gb
model_configs:
description: "Model config(s) - comma-separated (e.g. 'recipes/esm2_native_te_15b' or 'recipes/esm2_native_te_15b,recipes/esm2_native_te_3b'). Leave blank for all defaults."
required: false
default: ""
type: string
branch:
description: "Branch to use (ignored if commit SHA is provided)"
required: true
default: "main"
type: string
commit_sha:
description: "Commit SHA (optional - overrides branch if provided)"
required: false
type: string
schedule:
- cron: "0 8 * * *" # everyday at 1am PST
jobs:
submit-lepton-jobs:
runs-on: ubuntu-latest
strategy:
matrix:
model_config: >-
${{
github.event_name == 'schedule' && fromJSON('["esm2_native_te_650m", "esm2_native_te_3b", "esm2_native_te_15b"]') ||
github.event.inputs.model_configs != '' && fromJSON(format('["{0}"]', join(split(github.event.inputs.model_configs, ','), '","'))) ||
fromJSON('["esm2_native_te_650m", "esm2_native_te_3b", "esm2_native_te_15b"]')
}}
fail-fast: false
steps:
- name: Checkout
uses: actions/checkout@v4
with:
ref: ${{ github.event.inputs.commit_sha || github.event.inputs.branch || 'main' }}
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: "3.11"
cache: "pip"
cache-dependency-path: |
ci/lepton/requirements.txt
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r ci/lepton/requirements.txt
- name: Submit Lepton Jobs
env:
LEP_LOGIN_CREDENTIALS: ${{ secrets.LEP_LOGIN_CREDENTIALS }}
GPU_TYPE: ${{ github.event.inputs.gpu_type || 'h100-sxm' }}
BRANCH: ${{ github.event.inputs.branch || 'main' }}
COMMIT_SHA: ${{ github.event.inputs.commit_sha || '' }}
MODEL_CONFIG: ${{ matrix.model_config }}
run: |
set -euo pipefail
lep login -c "$LEP_LOGIN_CREDENTIALS" || true
# Map GPU type to node group
if [ "$GPU_TYPE" = "h200" ]; then
NODE_GROUP="nv-int-multiteam-nebius-h200-01"
elif [ "$GPU_TYPE" = "h100-sxm" ]; then
NODE_GROUP="yo-bom-lepton-001"
elif [ "$GPU_TYPE" = "a100-80gb" ]; then
NODE_GROUP="az-sat-lepton-001"
else
echo "Error: Unknown GPU type: $GPU_TYPE"
exit 1
fi
python ci/lepton/core/launch_job.py \
--config-path ../model_convergence/configs \
--config-name "recipes/$MODEL_CONFIG" \
branch="$BRANCH" \
commit_sha=$COMMIT_SHA \
node_group=$NODE_GROUP \
gpu_type="$GPU_TYPE"