Skip to content

Commit

Permalink
remove project id being passed into bigquery wrapper
Browse files Browse the repository at this point in the history
  • Loading branch information
reeba212 committed Oct 17, 2024
1 parent eabafaf commit 844d9de
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion sdks/python/apache_beam/io/gcp/bigquery_tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -352,7 +352,7 @@ class BigQueryWrapper(object):
HISTOGRAM_METRIC_LOGGER = MetricLogger()

def __init__(self, client=None, temp_dataset_id=None, temp_table_ref=None, project_id = None):
self.client = client or BigQueryWrapper._bigquery_client(PipelineOptions(project = project_id))
self.client = client or BigQueryWrapper._bigquery_client(PipelineOptions())
self.gcp_bq_client = client or gcp_bigquery.Client(
client_info=ClientInfo(
user_agent="apache-beam-%s" % apache_beam.__version__))
Expand Down
2 changes: 1 addition & 1 deletion sdks/python/apache_beam/yaml/integration_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,7 +65,7 @@ def temp_spanner_table(project, prefix='temp_spanner_db_'):

@contextlib.contextmanager
def temp_bigquery_table(project, prefix='yaml_bq_it_'):
bigquery_client = BigQueryWrapper(project_id = "apache-beam-testing")
bigquery_client = BigQueryWrapper()
dataset_id = '%s_%s' % (prefix, uuid.uuid4().hex)
bigquery_client.get_or_create_dataset(project, dataset_id)
logging.info("Created dataset %s in project %s", dataset_id, project)
Expand Down

0 comments on commit 844d9de

Please sign in to comment.