From f8f74bed710572c9b6835e5792d0a0fd5387c3e0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Alejandro=20Mart=C3=ADnez=20Otal?= <amartinezotal@gmail.com>
Date: Mon, 1 Jul 2024 23:18:47 +0200
Subject: [PATCH] :wrench: add timeout also for batch jobs in
 load_table_from_uri (#90)

Co-authored-by: Alejandro Martinez <alejandroa@kannactnow.com>
---
 target_bigquery/batch_job.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/target_bigquery/batch_job.py b/target_bigquery/batch_job.py
index 67f5305..bfa3f5c 100644
--- a/target_bigquery/batch_job.py
+++ b/target_bigquery/batch_job.py
@@ -63,6 +63,7 @@ def run(self) -> None:
                     BytesIO(job.data),
                     job.table,
                     num_retries=3,
+                    timeout=self.config.get("timeout", 600),
                     job_config=bigquery.LoadJobConfig(**job.config),
                 ).result()
             except Exception as exc: