add set_steps api in ps.py