Skip to content

Commit 958b537

Browse files
authored
chore: fix experimental blob display in partial ordering mode (#1484)
1 parent 00e0750 commit 958b537

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

bigframes/operations/blob.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ def display(self, n: int = 3, *, content_type: str = ""):
201201
content_type (str, default ""): content type of the blob. If unset, use the blob metadata of the storage. Possible values are "image", "audio" and "video".
202202
"""
203203
# col name doesn't matter here. Rename to avoid column name conflicts
204-
df = bigframes.series.Series(self._block).rename("blob_col").head(n).to_frame()
204+
df = bigframes.series.Series(self._block).rename("blob_col").to_frame()
205205

206206
df["read_url"] = df["blob_col"].blob.read_url()
207207

@@ -210,6 +210,9 @@ def display(self, n: int = 3, *, content_type: str = ""):
210210
else:
211211
df["content_type"] = df["blob_col"].blob.content_type()
212212

213+
pandas_df, _, query_job = df._block.retrieve_repr_request_results(n)
214+
df._set_internal_query_job(query_job)
215+
213216
def display_single_url(
214217
read_url: str, content_type: Union[str, pd._libs.missing.NAType]
215218
):
@@ -232,7 +235,7 @@ def display_single_url(
232235
response = requests.get(read_url)
233236
ipy_display.display(response.content)
234237

235-
for _, row in df.iterrows():
238+
for _, row in pandas_df.iterrows():
236239
display_single_url(row["read_url"], row["content_type"])
237240

238241
def _resolve_connection(self, connection: Optional[str] = None) -> str:

0 commit comments

Comments
 (0)