Merge pull request #432 from aai-institute/fix/cleanup #75

Job	Run time
Lint code and check type hints	1m 19s
Build Docs	1m 51s
Run notebook tests (3.8) / run-tests	20m 29s
Run notebook tests (3.10) / run-tests	19m 54s
Run notebook tests (3.9) / run-tests	23m 21s
Run tests that use PyTorch (3.10) / run-tests	12m 49s
Run tests that use PyTorch (3.8) / run-tests	10m 7s
Run tests that use PyTorch (3.9) / run-tests	15m 38s
Run base tests (3.10) / run-tests	23m 40s
Run base tests (3.8) / run-tests	27m 56s
Run base tests (3.9) / run-tests	25m 5s
Push Docs and maybe release Package to TestPyPI	1m 52s
	3h 4m 1s

Provide feedback