Skip to content

Commit

Permalink
revert changes that were addressed in other pull requests and just cr…
Browse files Browse the repository at this point in the history
…eate pr noise here
  • Loading branch information
mikealfare committed Dec 11, 2024
1 parent 6e27cbb commit 6179a63
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 2 deletions.
2 changes: 2 additions & 0 deletions dbt/adapters/bigquery/impl.py
Original file line number Diff line number Diff line change
Expand Up @@ -671,6 +671,7 @@ def load_dataframe(
table_name,
table_schema,
field_delimiter,
fallback_timeout=300,
)

@available.parse_none
Expand All @@ -691,6 +692,7 @@ def upload_file(
database,
table_schema,
table_name,
fallback_timeout=300,
**kwargs,
)

Expand Down
2 changes: 1 addition & 1 deletion dbt/adapters/bigquery/python_submissions.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ def __init__(self, parsed_model: Dict, credentials: BigQueryCredentials) -> None

# set retry policy, default to timeout after 24 hours
retry = RetryFactory(credentials)
self._polling_retry = retry.create_job_execution_polling(
self._polling_retry = retry.create_polling(
model_timeout=parsed_model["config"].get("timeout")
)

Expand Down
2 changes: 1 addition & 1 deletion dbt/adapters/bigquery/retry.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def __call__(self, error: Exception) -> bool:

def _create_reopen_on_error(connection: Connection) -> Callable[[Exception], None]:

def on_error(error: Exception) -> None:
def on_error(error: Exception):
if isinstance(error, (ConnectionResetError, ConnectionError)):
_logger.warning("Reopening connection after {!r}".format(error))
connection.handle.close()
Expand Down

0 comments on commit 6179a63

Please sign in to comment.