Skip to content

Commit 1c8d510

Browse files
chore: close session after partial ordering mode sample (#1173)
* 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>
1 parent be74b99 commit 1c8d510

File tree

1 file changed

+24
-19
lines changed

1 file changed

+24
-19
lines changed

samples/snippets/ordering_mode_partial_test.py

Lines changed: 24 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -16,30 +16,35 @@
1616
def test_quickstart() -> None:
1717
import bigframes.pandas
1818

19-
# We need a fresh session since we're modifying connection options.
20-
bigframes.pandas.close_session()
19+
try:
20+
# We need a fresh session since we're modifying connection options.
21+
bigframes.pandas.close_session()
2122

22-
# [START bigquery_bigframes_ordering_mode_partial]
23-
import bigframes.pandas as bpd
23+
# [START bigquery_bigframes_ordering_mode_partial]
24+
import bigframes.pandas as bpd
2425

25-
bpd.options.bigquery.ordering_mode = "partial"
26-
# [END bigquery_bigframes_ordering_mode_partial]
26+
bpd.options.bigquery.ordering_mode = "partial"
27+
# [END bigquery_bigframes_ordering_mode_partial]
2728

28-
# [START bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
29-
import warnings
29+
# [START bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
30+
import warnings
3031

31-
import bigframes.exceptions
32+
import bigframes.exceptions
3233

33-
warnings.simplefilter(
34-
"ignore", category=bigframes.exceptions.AmbiguousWindowWarning
35-
)
36-
# [END bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
34+
warnings.simplefilter(
35+
"ignore", category=bigframes.exceptions.AmbiguousWindowWarning
36+
)
37+
# [END bigquery_bigframes_ordering_mode_partial_ambiguous_window_warning]
3738

38-
df = bpd.DataFrame({"column": [1, 2, 1, 3, 1, 2, 3]})
39+
df = bpd.DataFrame({"column": [1, 2, 1, 3, 1, 2, 3]})
3940

40-
# [START bigquery_bigframes_ordering_mode_partial_drop_duplicates]
41-
# Avoid order dependency by using groupby instead of drop_duplicates.
42-
unique_col = df.groupby(["column"], as_index=False).size().drop(columns="size")
43-
# [END bigquery_bigframes_ordering_mode_partial_drop_duplicates]
41+
# [START bigquery_bigframes_ordering_mode_partial_drop_duplicates]
42+
# Avoid order dependency by using groupby instead of drop_duplicates.
43+
unique_col = df.groupby(["column"], as_index=False).size().drop(columns="size")
44+
# [END bigquery_bigframes_ordering_mode_partial_drop_duplicates]
4445

45-
assert len(unique_col) == 3
46+
assert len(unique_col) == 3
47+
finally:
48+
# Don't leak partial ordering mode to other code samples.
49+
bigframes.pandas.close_session()
50+
bpd.options.bigquery.ordering_mode = "strict"

0 commit comments

Comments
 (0)