Be explicit about python3 only support
Let the user know when we call ddp.wait()