fix: expose max_payload and assemble_with params (#19)

jina-ai · Feb 7, 2024 · fa9e2bf · fa9e2bf
1 parent d7a3d7b
commit fa9e2bf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/jina_sagemaker/client.py b/jina_sagemaker/client.py
@@ -136,6 +136,8 @@ def create_transform_job(
         role: Optional[str] = None,
         wait: bool = True,
         logs: bool = True,
+        assemble_with: Optional[str] = None,
+        max_payload: Optional[int] = None,
     ) -> Optional[str]:
         import sagemaker
 
@@ -182,6 +184,8 @@ def create_transform_job(
             instance_type=instance_type,
             output_path=output_path,
             strategy="MultiRecord",
+            assemble_with=assemble_with,
+            max_payload=max_payload,
         )
 
         transformer.transform(