fix singlestep dpm tests (#9716)

Commit

1 year ago

fix singlestep dpm tests (#9716) fix Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>