Skip to content
Run Unit Tests

This CL introduces an end-to-end Flax implementation of the DLRM-HSTU model, along with individual unit tests for the various components. #128

Sign in to view logs

This CL introduces an end-to-end Flax implementation of the DLRM-HSTU model, along with individual unit tests for the various components.

This CL introduces an end-to-end Flax implementation of the DLRM-HSTU model, along with individual unit tests for the various components. #128

Workflow file for this run

.github/workflows/unit_tests.yaml at d18dda9
name: Run Unit Tests
on:
push:
branches: [ "main", "master" ]
pull_request:
branches: [ "main", "master" ]
workflow_dispatch:
jobs:
run_tests:
name: Run Tests
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@v4
- name: Install Dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt # If you have a requirements.txt file
- name: Run Pytest
run: |
export KERAS_BACKEND=jax
python -m pytest -v