You shouldnt assign step method yourself unless you have special use case - try just trace = pm.sample()
trace = pm.sample()