update python wrapping class to new single step policy