Improve tolerance to non-array leaf types #80

	name: mypy

	on:
	push:
	branches: [main]
	pull_request:
	branches: [main]

	jobs:
	mypy:
	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: ["3.12"]

	steps:
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v1
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -e .
	pip install mypy
	- name: Test with mypy
	run: \|
	mypy .

Provide feedback