Skip to content

Commit 063e7ca

Browse files
committed
Batch transform: Add support for split_type == "None" in local mode
1 parent 41feb4c commit 063e7ca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/sagemaker/local/data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def get_splitter_instance(split_type):
6969
Returns
7070
:class:`sagemaker.local.data.Splitter`: an Instance of a Splitter
7171
"""
72-
if split_type is None:
72+
if split_type == "None" or split_type is None:
7373
return NoneSplitter()
7474
if split_type == "Line":
7575
return LineSplitter()

0 commit comments

Comments
 (0)