Updates custom driver to include custom_url_partitioner