Skip to content

Commit b4cdb05

Browse files
matthew29tangcopybara-github
authored andcommitted
fix: Fix Bigframes tensorflow serializer dependencies
PiperOrigin-RevId: 574226794
1 parent 598d57d commit b4cdb05

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

vertexai/preview/_workflow/serialization_engine/serializers.py

+2
Original file line numberDiff line numberDiff line change
@@ -1157,7 +1157,9 @@ def serialize(
11571157
BigframeSerializer._metadata.custom_commands.append("pip install torcharrow")
11581158
elif detected_framework == "tensorflow":
11591159
tensorflow_io_dep = "tensorflow-io==" + self._get_tfio_verison()
1160+
tensorflow_io_gcs_fs_dep = "tensorflow-io-gcs-filesystem==" + self._get_tfio_verison()
11601161
BigframeSerializer._metadata.dependencies.append(tensorflow_io_dep)
1162+
BigframeSerializer._metadata.dependencies.append(tensorflow_io_gcs_fs_dep)
11611163

11621164
# Check if index.name is default and set index.name if not
11631165
if to_serialize.index.name and to_serialize.index.name != "index":

0 commit comments

Comments
 (0)