Skip to content

Commit

Permalink
Removing type hint
Browse files Browse the repository at this point in the history
  • Loading branch information
benfhicks committed Nov 20, 2024
1 parent ec44040 commit 50e5fe9
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions mlops_tooling/bq_connector/bigquery_connector.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import re
from pathlib import Path
from typing import Generator

import pandas as pd
from google.cloud import bigquery
Expand Down Expand Up @@ -97,7 +96,7 @@ def query(self, sql_file: str, create_bqstorage_client: bool = False, **kwargs):

def query_in_chunks(
self, sql_file: str, chunk_size: int = 10000, schema: list[str] = [], **kwargs
) -> Generator[pd.DataFrame | list]:
):
"""
Returns all values from a table and outputs results in chunks.
Expand Down

0 comments on commit 50e5fe9

Please sign in to comment.