Проблема с pytest-xdist при перезагрузке ноды

Добрый день, коллеги!

Рассматриваю в качестве инструмента автоматизации pytest+xdist, нужно запускать набор тестов на 20 ОС Linux параллельно, но столкнулся с ограничением, которое ставит под вопрос использование инструмента.
Есть набор тестов, где нужна перезагрузка ноды, в xdist, как я понял, нельзя задать timeout, чтобы он стучался по ssh к ноде в течении него, сейчас тест просто валится с потерей связи. Может есть какой то workaround, кто сталкивался напишите пожалуйста. Спасибо!