So I think there are two issues here (1) the parameters piece feels like a order of execution thing and should be simple to solve (2) we're a little sceptical about the mlflow hook never firing, could you put a breakpoint/logging the prove this