Skip to content

Commit 4825aeb

Browse files
committed
fix: fix mypy
1 parent fd04e6a commit 4825aeb

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

bigframes/display/html.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -230,15 +230,18 @@ def repr_mimebundle(
230230
f"Falling back to static HTML. Error: {traceback.format_exc()}"
231231
)
232232

233+
blob_cols: list[str]
233234
if isinstance(obj, bigframes.series.Series):
234-
df = obj
235-
blob_cols: list[str] = []
235+
pandas_df, row_count, query_job = obj._block.retrieve_repr_request_results(
236+
opts.max_rows
237+
)
238+
blob_cols = []
236239
else:
237240
df, blob_cols = obj._get_display_df_and_blob_cols()
241+
pandas_df, row_count, query_job = df._block.retrieve_repr_request_results(
242+
opts.max_rows
243+
)
238244

239-
pandas_df, row_count, query_job = df._block.retrieve_repr_request_results(
240-
opts.max_rows
241-
)
242245
obj._set_internal_query_job(query_job)
243246
column_count = len(pandas_df.columns)
244247

0 commit comments

Comments
 (0)