Подключение к CloudSQL из потока данных в Python

Я пытаюсь подключиться к CloudSQL с помощью конвейера Python.

Актуальная ситуация

• Я могу сделать это без проблем с помощью DirectRunner
• Не могу подключиться с помощью DataflowRunner

Функция подключения

def cloudSQL(input):
    import pymysql
    connection = pymysql.connect(host='<server ip>',
                                   user='...',
                                   password='...',
                                   db='...')
    cursor = connection.cursor()
    cursor.execute("select ...")
    connection.close()
    result = cursor.fetchone()
    if not (result is None):
        yield input

Ошибка

Это сообщение об ошибке при использовании DataflowRunner

OperationalError: (2003, "Can't connect to MySQL server on '<server ip>' (timed out)")

CloudSQL

У меня есть publicIP (для тестирования с локального с помощью directrunner), и я также пытался активировать частный IP, чтобы узнать, может ли это быть проблемой для подключения к DataflowRunner

Вариант 2

Я также пробовал с

connection = pymysql.connect((unix_socket='/cloudsql/' + <INSTANCE_CONNECTION_NAME>,
                               user='...',
                               password='...',
                               db='...')

С ошибкой:

OperationalError: (2003, "Can't connect to MySQL server on 'localhost' ([Errno 2] No such file or directory)")