Skip to content

Commit caf4948

Browse files
committed
fix test_blob_image_resize* and test_blob_image_normalize*
1 parent 8fb07c9 commit caf4948

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

bigframes/operations/blob.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -489,7 +489,12 @@ def image_blur(
489489
ops.JSONValue(json_path="$.status")
490490
)
491491
results_df = bpd.DataFrame(
492-
{"status": blurred_status_series, "content": dst}
492+
{
493+
"status": blurred_status_series,
494+
"content": dst.blob.uri().str.to_blob(
495+
connection=self._resolve_connection(connection)
496+
),
497+
}
493498
)
494499
results_struct = bbq.struct(results_df).rename("blurred_results")
495500
return results_struct
@@ -623,7 +628,12 @@ def image_resize(
623628
ops.JSONValue(json_path="$.status")
624629
)
625630
results_df = bpd.DataFrame(
626-
{"status": resized_status_series, "content": dst}
631+
{
632+
"status": resized_status_series,
633+
"content": dst.blob.uri().str.to_blob(
634+
connection=self._resolve_connection(connection)
635+
),
636+
}
627637
)
628638
results_struct = bbq.struct(results_df).rename("resized_results")
629639
return results_struct
@@ -753,7 +763,9 @@ def image_normalize(
753763
results_df = bpd.DataFrame(
754764
{
755765
"status": normalized_status_series,
756-
"content": dst,
766+
"content": dst.blob.uri().str.to_blob(
767+
connection=self._resolve_connection(connection)
768+
),
757769
}
758770
)
759771
results_struct = bbq.struct(results_df).rename("normalized_results")

0 commit comments

Comments
 (0)