support pytrees as diag_shift (#1135)

* support pytrees as diag_shift

* add test

* adapt test for dense and pytree
(problem is finding the correct shape of shift when e.g. doing real-imag 
split; ideally we should convert it in the constructor of the qgt)

* black

* isinstance instead of checking attr