Python wrapper for TreeCollection