Skip to content

Commit 512b82d

Browse files
matthew29tangcopybara-github
authored andcommitted
fix: Fix crash when no target_col is specified for Bigframes tensorflow
PiperOrigin-RevId: 602543376
1 parent b9b373b commit 512b82d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vertexai/preview/_workflow/serialization_engine/serializers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1284,7 +1284,7 @@ def _deserialize_tensorflow(
12841284
"""
12851285
# Set default kwarg values
12861286
batch_size = batch_size or DEFAULT_TENSORFLOW_BATCHSIZE
1287-
target_col = target_col.encode("ASCII") or b"target"
1287+
target_col = target_col.encode("ASCII") if target_col else b"target"
12881288

12891289
# Deserialization at remote environment
12901290
try:

0 commit comments

Comments
 (0)