fix: Fix Bigframes tensorflow serializer dependencies

PiperOrigin-RevId: 574226794
googleapis · Oct 17, 2023 · b4cdb05 · b4cdb05
1 parent 598d57d
commit b4cdb05
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/vertexai/preview/_workflow/serialization_engine/serializers.py b/vertexai/preview/_workflow/serialization_engine/serializers.py
@@ -1157,7 +1157,9 @@ def serialize(
             BigframeSerializer._metadata.custom_commands.append("pip install torcharrow")
         elif detected_framework == "tensorflow":
             tensorflow_io_dep = "tensorflow-io==" + self._get_tfio_verison()
+            tensorflow_io_gcs_fs_dep = "tensorflow-io-gcs-filesystem==" + self._get_tfio_verison()
             BigframeSerializer._metadata.dependencies.append(tensorflow_io_dep)
+            BigframeSerializer._metadata.dependencies.append(tensorflow_io_gcs_fs_dep)
 
         # Check if index.name is default and set index.name if not
         if to_serialize.index.name and to_serialize.index.name != "index":