File tree 1 file changed +3
-2
lines changed
vertexai/preview/_workflow/serialization_engine
1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -1146,8 +1146,9 @@ def serialize(
1146
1146
detected_framework = kwargs .get ("framework" )
1147
1147
BigframeSerializer ._metadata .framework = detected_framework
1148
1148
if detected_framework == "torch" :
1149
- self .register_custom_command ("pip install torchdata" )
1150
- self .register_custom_command ("pip install torcharrow" )
1149
+ # Install using custom_commands to avoid numpy dependency conflict
1150
+ BigframeSerializer ._metadata .custom_commands .append ("pip install torchdata" )
1151
+ BigframeSerializer ._metadata .custom_commands .append ("pip install torcharrow" )
1151
1152
elif detected_framework == "tensorflow" :
1152
1153
tensorflow_io_dep = "tensorflow-io==" + self ._get_tfio_verison ()
1153
1154
BigframeSerializer ._metadata .dependencies .append (tensorflow_io_dep )
You can’t perform that action at this time.
0 commit comments