None of that worked. And I actually wouldn’t be happy if it did, because the original code is inside a class. The code I pasted was just to simplify things as a minimal working example. So (I hope) it shouldn’t matter where the pm.sample is called from.
This is very weird. I have another model (parameterization) that works just fine in a very similar code structure.