BQML и TensorFlow: время выполнения UDF-worker истекло; Произошло неожиданное прерывание для рабочего tensorflow-worker-166: request_timeout

Question

У меня сохранена модель TensorFlow, и я успешно создал из нее модель BigQuery ML.

Я могу успешно вызвать модель и получить результаты по небольшому количеству строк (примерно до 50 или около того), выполнив что-то вроде этого:

WITH model_input AS (
  SELECT * FROM `my_dataset.my_input_table`
  ORDER BY RAND()
  LIMIT 10
)
SELECT
    *
  FROM
    ML.PREDICT(MODEL `my_dataset.my_model`, table model_input)

но если я удалю предложение о ограничении и попытаюсь запустить его для всего набора данных (около 7000 строк), произойдет сбой с таким сообщением об ошибке:

UDF worker timed out during execution.; Unexpected abort triggered for worker tensorflow-worker-166: request_timeout

Есть ли способ запустить свой запрос без ручного написания сценария какой-либо пакетной операции?

ОБНОВИТЬ:

Я должен уточнить, что это не проблема с конкретной строкой. Я могу получить те же 100 строк в 2 пакетах по 50, которые вызывают ошибку, если я попытаюсь сделать это одним пакетом из 100