Transitioning from pm.sample() to pm.fit()

Will follow up if I find something that works :+1: