Fix Jax breakage where assert_allclose fail for tf/torch tensor #2424

	name: Unittests & Auto-publish

	# Allow to trigger the workflow manually (e.g. when deps changes)
	on: [push, workflow_dispatch]

	jobs:
	pytest-job:
	strategy:
	fail-fast: false
	matrix:
	python-version: ['3.10', '3.11', '3.12']
	runs-on: ubuntu-latest
	timeout-minutes: 30

	concurrency:
	group: ${{ github.workflow }}-${{ github.ref }}-${{ matrix.python-version }}
	cancel-in-progress: true

	steps:
	- uses: actions/checkout@v3

	# Required to test `mediapy`
	- run: sudo apt-get update
	- run: sudo apt-get install -y ffmpeg

	# Install deps
	- uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}
	- run: pip --version
	- run: pip install -e .[all,dev]
	- run: pip freeze

	# Run tests
	- name: Run core tests
	run: pytest -vv -n auto

	# Auto-publish when version is increased
	publish-job:
	# Only try to publish if:
	# * Repo is self (prevents running from forks)
	# * Branch is `main`
	if: \|
	github.repository == 'google/etils'
	&& github.ref == 'refs/heads/main'
	needs: pytest-job # Only publish after tests are successful
	runs-on: ubuntu-latest
	permissions:
	contents: write
	timeout-minutes: 30

	steps:
	# Publish the package (if local `__version__` > pip version)
	- uses: etils-actions/pypi-auto-publish@v1
	with:
	pypi-token: ${{ secrets.PYPI_API_TOKEN }}
	gh-token: ${{ secrets.GITHUB_TOKEN }}
	parse-changelog: true

Provide feedback