Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions samples/snippets/sessions_and_io_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@

def test_sessions_and_io(project_id: str, dataset_id: str, gcs_bucket: str) -> None:
YOUR_PROJECT_ID = project_id
YOUR_DATASET_ID = dataset_id
YOUR_LOCATION = "us"
YOUR_BUCKET = gcs_bucket

Expand Down Expand Up @@ -66,7 +67,7 @@ def test_sessions_and_io(project_id: str, dataset_id: str, gcs_bucket: str) -> N
# [END bigquery_dataframes_set_options_for_global_session]

# [START bigquery_dataframes_global_session_is_the_default_session]
# The following two statements are essentiall the same
# The following two statements are essentially the same
df = bpd.read_gbq("bigquery-public-data.ml_datasets.penguins")
df = bpd.get_global_session().read_gbq("bigquery-public-data.ml_datasets.penguins")
# [END bigquery_dataframes_global_session_is_the_default_session]
Expand Down Expand Up @@ -168,12 +169,12 @@ def test_sessions_and_io(project_id: str, dataset_id: str, gcs_bucket: str) -> N
# [END bigquery_dataframes_read_from_sql_query]
assert df is not None

table_name = "snippets-session-and-io-test"
YOUR_TABLE_NAME = "snippets-session-and-io-test"

# [START bigquery_dataframes_dataframe_to_bigquery_table]
import bigframes.pandas as bpd

df = bpd.DataFrame({"my_col": [1, 2, 3]})

df.to_gbq(f"{project_id}.{dataset_id}.{table_name}")
df.to_gbq(f"{YOUR_PROJECT_ID}.{YOUR_DATASET_ID}.{YOUR_TABLE_NAME}")
# [END bigquery_dataframes_dataframe_to_bigquery_table]