Move build-latest CI job to upstream tests #79

	name: Test

	on:
	# Trigger the workflow on push or pull request, but only on main branch
	push:
	branches:
	- main
	pull_request:
	branches:
	- main

	permissions:
	contents: read # to fetch code

	concurrency:
	group: ${{ github.workflow }}-${{ github.head_ref \|\| github.ref }}
	cancel-in-progress: true

	# TODO(jakevdp): add testing on macOS-11 and windows-2019 when grain supports them,
	# or alternatively run a subset of tests without grain.
	jobs:
	build:
	name: Pinned packages (${{ matrix.os }} Python ${{ matrix.python-version }})
	runs-on: ${{ matrix.os }}
	timeout-minutes: 10
	strategy:
	matrix:
	os: ["ubuntu-latest"]
	python-version: ["3.10", "3.11", "3.12"]

	steps:
	- uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332 # v4.1.7
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@f677139bbe7f9c59b41e40162b753c062f5d49a3 # v5.2.0
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install .[dev,tfds,grain]
	- name: Run tests
	run: \|
	pytest -n auto jax_ai_stack

Provide feedback