Flax/nnx backend #317

	name: Integration Tests
	on:
	pull_request:
	push:
	branches:
	- main

	jobs:
	integration-tests:
	name: Run Integration Tests
	runs-on: ubuntu-latest
	strategy:
	matrix:
	# Select the Python versions to test against
	os: ["ubuntu-latest", "macos-latest"]
	python-version: ["3.10", "3.11"]
	fail-fast: true
	steps:
	- name: Check out the code
	uses: actions/[email protected]
	with:
	fetch-depth: 1
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v4
	with:
	python-version: ${{ matrix.python-version }}

	# Install Poetry
	- name: Install Poetry
	uses: snok/[email protected]
	with:
	version: 1.4.0

	# Configure Poetry to use the virtual environment in the project
	- name: Setup Poetry
	run: \|
	poetry config virtualenvs.in-project true

	# Install the dependencies
	- name: Install Package
	run: \|
	poetry install --all-extras --with docs

	# Run the unit tests and build the coverage report
	- name: Run Integration Tests
	run: poetry run python tests/integration_tests.py

Provide feedback