From 512b82debdaf44d1b9c755c2bd06d1d24c080338 Mon Sep 17 00:00:00 2001 From: Matthew Tang Date: Mon, 29 Jan 2024 17:07:12 -0800 Subject: [PATCH] fix: Fix crash when no target_col is specified for Bigframes tensorflow PiperOrigin-RevId: 602543376 --- vertexai/preview/_workflow/serialization_engine/serializers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vertexai/preview/_workflow/serialization_engine/serializers.py b/vertexai/preview/_workflow/serialization_engine/serializers.py index 6886a47565..7a04a0eb92 100644 --- a/vertexai/preview/_workflow/serialization_engine/serializers.py +++ b/vertexai/preview/_workflow/serialization_engine/serializers.py @@ -1284,7 +1284,7 @@ def _deserialize_tensorflow( """ # Set default kwarg values batch_size = batch_size or DEFAULT_TENSORFLOW_BATCHSIZE - target_col = target_col.encode("ASCII") or b"target" + target_col = target_col.encode("ASCII") if target_col else b"target" # Deserialization at remote environment try: