This is a great point and is something we're planning on adding in an upcoming release of our Python library.