geweke takes a 1D trace of some (single) parameter. So try something like this:
pm.geweke(linear_trace['Beta0'])