Move all NNX content up a level to be equal with Linen, to make python packaging more consistent. #371

	name: CI

	on:
	# Triggers the workflow on push or pull request events but only for the main branch
	push:
	branches: [ main ]
	pull_request:
	branches: [ main ]
	# Allows you to run this workflow manually from the Actions tab
	workflow_dispatch:

	jobs:
	build:
	name: "python ${{ matrix.python-version }}"
	runs-on: ubuntu-latest
	strategy:
	fail-fast: false
	matrix:
	python-version: ["3.10", "3.11"]
	steps:
	- name: Cancel previous
	uses: styfle/[email protected]
	with:
	access_token: ${{ github.token }}
	if: ${{github.ref != 'refs/head/main'}}
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v2
	with:
	python-version: ${{ matrix.python-version }}
	- name: Get pip cache dir
	id: pip-cache
	run: \|
	python -m pip install --upgrade pip wheel
	echo "::set-output name=dir::$(pip cache dir)"
	- name: pip cache
	uses: actions/cache@v2
	with:
	path: ${{ steps.pip-cache.outputs.dir }}
	key: ${{ runner.os }}-pip-${{ hashFiles('**/pyproject.toml') }}
	- name: Install NeuralGCM
	run: \|
	pip install -e .[tests]
	- name: Run unit tests
	run: \|
	pytest -v neuralgcm

Provide feedback