Skip to content
Pytest

Remove embed_* variants of model architectures in speculator training #162

Sign in to view logs

Remove embed_* variants of model architectures in speculator training

Remove embed_* variants of model architectures in speculator training #162

Workflow file for this run

.github/workflows/pytest.yml at 68dfaf6
# This workflow will install dependencies and run pytest
name: Pytest
on: [pull_request]
permissions:
contents: read
jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python 3.10
id: setup_python
uses: actions/setup-python@v5
with:
python-version: "3.10"
- name: Restore Virtualenv
uses: actions/cache/restore@v4
id: cache-venv-restore
with:
path: ./.venv/
key: ${{ runner.os }}-${{ steps.setup_python.outputs.python-version }}-venv-${{ hashFiles('*requirements.txt') }}
restore-keys: |
${{ runner.os }}-${{ steps.setup_python.outputs.python-version }}-venv-
- name: Install dependencies
run: |
# Create the virtual environment
python -m venv .venv
. ./.venv/bin/activate
# Install the dependencies
# In case of a cache hit on the primary key, this will be a no-op
# In case of a cache miss, but hit on a secondary key, this will update what's changed
python -m pip install --upgrade pip
pip install -r test-requirements.txt
# Enables the virtual env for following steps
echo "$VIRTUAL_ENV/bin" >> $GITHUB_PATH
echo "VIRTUAL_ENV=$VIRTUAL_ENV" >> $GITHUB_ENV
- name: Test with pytest
run: |
# Install ibm-fms from the main branch for testing purposes
# Use -I to ignore the existing install and actually install
# the version on main
pip install -I ibm-fms@git+https://github.com/foundation-model-stack/foundation-model-stack@main
# Install fms-fsdp project
pip install -e .
# No type stubs available for "fire" and "transformers"
pytest tests/
- name: Save Virtualenv
id: cache-venv-save
uses: actions/cache/save@v4
with:
path: ./.venv/
key: ${{ steps.cache-venv-restore.outputs.cache-primary-key }}