fix: Fix crash when no target_col is specified for Bigframes tensorflow

PiperOrigin-RevId: 602543376
googleapis · Jan 30, 2024 · 512b82d · 512b82d
1 parent b9b373b
commit 512b82d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: