Change None to no_op in SDM Trainer