File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
plugins/flytekit-openai/flytekitplugins/openai/batch Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -39,7 +39,7 @@ class JSONIteratorTransformer(TypeTransformer[Iterator[JSON]]):
39
39
A JSON iterator that handles conversion between an iterator/generator and a JSONL file.
40
40
"""
41
41
42
- JSON_ITERATOR_FORMAT = "JSONL "
42
+ JSON_ITERATOR_FORMAT = "jsonl "
43
43
44
44
def __init__ (self ):
45
45
super ().__init__ ("JSON Iterator" , Iterator [JSON ])
Original file line number Diff line number Diff line change @@ -89,7 +89,7 @@ def __init__(
89
89
self ,
90
90
name : str ,
91
91
task_config : OpenAIFileConfig ,
92
- container_image : str = OpenAIFileDefaultImages .default_image (),
92
+ container_image : str = OpenAIFileDefaultImages .find_image_for (),
93
93
** kwargs ,
94
94
):
95
95
super ().__init__ (
@@ -140,7 +140,7 @@ def __init__(
140
140
self ,
141
141
name : str ,
142
142
task_config : OpenAIFileConfig ,
143
- container_image : str = OpenAIFileDefaultImages .default_image (),
143
+ container_image : str = OpenAIFileDefaultImages .find_image_for (),
144
144
** kwargs ,
145
145
):
146
146
super ().__init__ (
You can’t perform that action at this time.
0 commit comments