model.py fromVertexAi: Add GRPC payload formats to the paylaodFormat …

…documentation. PiperOrigin-RevId: 682375728
google · Oct 4, 2024 · 4ef76e6 · 4ef76e6
1 parent f2e8070
commit 4ef76e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/ee/model.py b/python/ee/model.py
@@ -227,7 +227,8 @@ def fromVertexAi(
       maxPayloadBytes: The prediction payload size limit in bytes. Defaults to
         1.5MB (1500000 bytes.)
       payloadFormat: The payload format of entries in prediction requests and
-        responses. One of: ['SERIALIZED_TF_TENSORS, 'RAW_JSON', 'ND_ARRAYS'].
+        responses. One of: ['SERIALIZED_TF_TENSORS, 'RAW_JSON', 'ND_ARRAYS',
+        'GRPC_TF_TENSORS', 'GRPC_SERIALIZED_TF_TENSORS', 'GRPC_TF_EXAMPLES'].
         Defaults to 'SERIALIZED_TF_TENSORS'.
 
     Returns: