Skip to content

Commit 6d2f5a2

Browse files
committed
polish code
1 parent 00dd022 commit 6d2f5a2

File tree

1 file changed

+2
-4
lines changed
  • bigframes/bigquery/_operations

1 file changed

+2
-4
lines changed

bigframes/bigquery/_operations/ai.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
import pandas as pd
2525

2626
from bigframes import clients, dtypes, series, session
27-
from bigframes.core import convert, global_session, log_adapter
27+
from bigframes.core import convert, log_adapter
2828
from bigframes.operations import ai_ops, output_schemas
2929

3030
PROMPT_TYPE = Union[
@@ -519,9 +519,7 @@ def _separate_context_and_series(
519519
raise ValueError(f"Unsupported prompt type: {type(prompt)}")
520520

521521
if isinstance(prompt, str):
522-
return [None], [
523-
series.Series([prompt], session=global_session.get_global_session())
524-
]
522+
return [None], [series.Series([prompt])]
525523

526524
if isinstance(prompt, series.Series):
527525
if prompt.dtype == dtypes.OBJ_REF_DTYPE:

0 commit comments

Comments
 (0)