fix singlestep dpm tests (#9716)

Commit

2 years ago

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