Split Jax training dependencies (especially Scenic and TF-IO) into a separate dependency group. #8

	name: Build

	on: [push, pull_request]

	jobs:
	build_wheels:
	name: Build wheels on ${{ matrix.os }}
	runs-on: ${{ matrix.os }}
	strategy:
	matrix:
	os: [ubuntu-latest]

	steps:
	- uses: actions/checkout@v4

	- name: Set up Poetry
	run: pipx install poetry

	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	cache: 'poetry'

	- name: Install cibuildwheel
	run: python -m pip install cibuildwheel==2.16.5

	- name: Build wheels
	run: python -m cibuildwheel --output-dir wheelhouse
	# to supply options, put them in 'env', like:
	# env:
	# CIBW_SOME_OPTION: value

	- uses: actions/upload-artifact@v4
	with:
	name: cibw-wheels-${{ matrix.os }}-${{ strategy.job-index }}
	path: ./wheelhouse/*.whl

Provide feedback