convert jsonl to lowercase and update default image (#2438)

Signed-off-by: Samhita Alla <[email protected]>
flyteorg · May 22, 2024 · 4fcfe71 · 4fcfe71
1 parent d89181f
commit 4fcfe71
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/flytekit/types/iterator/json_iterator.py b/flytekit/types/iterator/json_iterator.py
@@ -39,7 +39,7 @@ class JSONIteratorTransformer(TypeTransformer[Iterator[JSON]]):
     A JSON iterator that handles conversion between an iterator/generator and a JSONL file.
     """
 
-    JSON_ITERATOR_FORMAT = "JSONL"
+    JSON_ITERATOR_FORMAT = "jsonl"
 
     def __init__(self):
         super().__init__("JSON Iterator", Iterator[JSON])

diff --git a/plugins/flytekit-openai/flytekitplugins/openai/batch/task.py b/plugins/flytekit-openai/flytekitplugins/openai/batch/task.py
@@ -89,7 +89,7 @@ def __init__(
         self,
         name: str,
         task_config: OpenAIFileConfig,
-        container_image: str = OpenAIFileDefaultImages.default_image(),
+        container_image: str = OpenAIFileDefaultImages.find_image_for(),
         **kwargs,
     ):
         super().__init__(
@@ -140,7 +140,7 @@ def __init__(
         self,
         name: str,
         task_config: OpenAIFileConfig,
-        container_image: str = OpenAIFileDefaultImages.default_image(),
+        container_image: str = OpenAIFileDefaultImages.find_image_for(),
         **kwargs,
     ):
         super().__init__(