Improve wandb logging #2324

Job	Run time
Run tests on ubuntu-latest with Python 3.10	1m 12s
Run tests on ubuntu-latest with Python 3.10	1h 0m 14s
Run tests on ubuntu-latest with Python 3.10	1h 23m 15s
Run tests on ubuntu-latest with Python 3.10	31m 17s
Run tests on ubuntu-latest with Python 3.10	33m 33s
Run tests on ubuntu-latest with Python 3.10	41m 27s
Run tests on ubuntu-latest with Python 3.10	34m 5s
Run tests on ubuntu-latest with Python 3.10	1m 39s
Run tests on ubuntu-latest with Python 3.10	42m 49s
Run tests on ubuntu-latest with Python 3.10	1h 4m 57s
Run tests on ubuntu-latest with Python 3.10	31m 46s
Run tests on ubuntu-latest with Python 3.10	31m 18s
Run tests on ubuntu-latest with Python 3.10	39m 20s
Run tests on ubuntu-latest with Python 3.10	33m 4s
Run tests on ubuntu-latest with Python 3.12	1m 46s
Run tests on ubuntu-latest with Python 3.12	42m 23s
Run tests on ubuntu-latest with Python 3.12	1h 3m 28s
Run tests on ubuntu-latest with Python 3.12	27m 56s
Run tests on ubuntu-latest with Python 3.12	29m 47s
Run tests on ubuntu-latest with Python 3.12	38m 47s
Run tests on ubuntu-latest with Python 3.12	30m 21s
	12h 44m 24s

Provide feedback