fix pyamg parametersweep