A bit of cleanup #63

Job	Run time
Build Docs	1m 55s
Lint code and check type hints	1m 53s
Run base tests (3.10) / run-tests	25m 6s
Run base tests (3.8) / run-tests	23m 2s
Run base tests (3.9) / run-tests	24m 36s
Run tests that use PyTorch (3.10) / run-tests	11m 27s
Run notebook tests (3.10) / run-tests	15m 4s
Run tests that use PyTorch (3.8) / run-tests	14m 18s
Run notebook tests (3.8) / run-tests	18m 16s
Run tests that use PyTorch (3.9) / run-tests	11m 46s
Run notebook tests (3.9) / run-tests	19m 43s
Push Docs and maybe release Package to TestPyPI	0s
	2h 47m 6s

Provide feedback