Skip to content

Commit

Permalink
chore: close session after partial ordering mode sample (#1173)
Browse files Browse the repository at this point in the history
* chore: close session after partial ordering mode sample

* 🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

* try manually selecting strict mode

---------

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
  • Loading branch information
tswast and gcf-owl-bot[bot] authored Nov 26, 2024
1 parent be74b99 commit 1c8d510
Showing 1 changed file with 24 additions and 19 deletions.
43 changes: 24 additions & 19 deletions samples/snippets/ordering_mode_partial_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,30 +16,35 @@
def test_quickstart() -> None:
import bigframes.pandas

# We need a fresh session since we're modifying connection options.
bigframes.pandas.close_session()
try:
# We need a fresh session since we're modifying connection options.
bigframes.pandas.close_session()

# [START bigquery_bigframes_ordering_mode_partial]
import bigframes.pandas as bpd
# [START bigquery_bigframes_ordering_mode_partial]
import bigframes.pandas as bpd

bpd.options.bigquery.ordering_mode = "partial"
# [END bigquery_bigframes_ordering_mode_partial]
bpd.options.bigquery.ordering_mode = "partial"
# [END bigquery_bigframes_ordering_mode_partial]

# [START bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
import warnings
# [START bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
import warnings

import bigframes.exceptions
import bigframes.exceptions

warnings.simplefilter(
"ignore", category=bigframes.exceptions.AmbiguousWindowWarning
)
# [END bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
warnings.simplefilter(
"ignore", category=bigframes.exceptions.AmbiguousWindowWarning
)
# [END bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]

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

# [START bigquery_bigframes_ordering_mode_partial_drop_duplicates]
# Avoid order dependency by using groupby instead of drop_duplicates.
unique_col = df.groupby(["column"], as_index=False).size().drop(columns="size")
# [END bigquery_bigframes_ordering_mode_partial_drop_duplicates]
# [START bigquery_bigframes_ordering_mode_partial_drop_duplicates]
# Avoid order dependency by using groupby instead of drop_duplicates.
unique_col = df.groupby(["column"], as_index=False).size().drop(columns="size")
# [END bigquery_bigframes_ordering_mode_partial_drop_duplicates]

assert len(unique_col) == 3
assert len(unique_col) == 3
finally:
# Don't leak partial ordering mode to other code samples.
bigframes.pandas.close_session()
bpd.options.bigquery.ordering_mode = "strict"

0 comments on commit 1c8d510

Please sign in to comment.