Update docstrings #517

	name: build

	env:
	PYTHON_MAIN_VERSION: 3.9

	on:
	pull_request:
	branches:
	- '*'
	push:
	branches:
	- '*'
	tags:
	- '*'

	jobs:
	build-linux:
	runs-on: ubuntu-latest
	strategy:
	max-parallel: 5
	matrix:
	python-version: ['3.9', '3.10']

	steps:
	- uses: actions/checkout@v2

	- name: Set up Python ${{ matrix.python-version }}

	uses: actions/setup-python@v2
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	sudo apt-get update -qq
	python -m pip install --upgrade pip
	python -m pip install flake8 pytest
	python -m pip install jaxlib
	python -m pip install jax
	if [ -f requirements.txt ]; then pip install -r requirements.txt; fi

	- name: install package
	run: \|
	pip install .
	pip list

	- name: Lint with flake8
	run: \|
	# stop the build if there are Python syntax errors or undefined names
	flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
	# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
	flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics

	- name: Generate coverage report
	run: \|
	pip install pytest
	pip install pytest-cov
	pytest --cov=./ --cov-report=xml

	- name: Upload coverage to Codecov
	if: ${{ matrix.python-version == env.PYTHON_MAIN_VERSION }}
	uses: codecov/codecov-action@v1
	with:
	token: ${{ secrets.CODECOV_TOKEN }}
	files: ./coverage.xml
	# directory: ./coverage/reports/
	flags: unittests
	env_vars: OS,PYTHON
	name: codecov-umbrella
	fail_ci_if_error: true
	# path_to_write_report: ./coverage/codecov_report.txt
	verbose: true

Provide feedback