Skip to content

Commit 628d8b8

Browse files
committed
Changed the way it loads the client
1 parent 24fe727 commit 628d8b8

File tree

1 file changed

+4
-1
lines changed
  • airflow-cloud-composer-etl-feature-train/materializers

1 file changed

+4
-1
lines changed

airflow-cloud-composer-etl-feature-train/materializers/bq_dataset.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,10 @@ def __init__(
3535
self.write_disposition = write_disposition
3636
self.project = project
3737
self.dataset = dataset
38-
self.client = bigquery.Client(project=project)
38+
if project is None:
39+
self.client = bigquery.Client()
40+
else:
41+
self.client = bigquery.Client(project=project)
3942
self.df = df
4043

4144
def read_data(self) -> pd.DataFrame:

0 commit comments

Comments
 (0)