Skip to content

Commit f32a441

Browse files
apackerlaurenyu
authored andcommitted
Add TFRecord split type to docs (#577)
1 parent c62eb02 commit f32a441

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sagemaker/transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def transform(self, data, data_type='S3Prefix', content_type=None, compression_t
9393
compression_type (str): Compression type of the input data, if compressed (default: None).
9494
Valid values: 'Gzip', None.
9595
split_type (str): The record delimiter for the input object (default: 'None').
96-
Valid values: 'None', 'Line', and 'RecordIO'.
96+
Valid values: 'None', 'Line', 'RecordIO', and 'TFRecord'.
9797
job_name (str): job name (default: None). If not specified, one will be generated.
9898
"""
9999
local_mode = self.sagemaker_session.local_mode

src/sagemaker/workflow/airflow.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -424,7 +424,7 @@ def transform_config(transformer, data, data_type='S3Prefix', content_type=None,
424424
compression_type (str): Compression type of the input data, if compressed (default: None).
425425
Valid values: 'Gzip', None.
426426
split_type (str): The record delimiter for the input object (default: 'None').
427-
Valid values: 'None', 'Line', and 'RecordIO'.
427+
Valid values: 'None', 'Line', 'RecordIO', and 'TFRecord'.
428428
job_name (str): job name (default: None). If not specified, one will be generated.
429429
430430
Returns:
@@ -495,7 +495,7 @@ def transform_config_from_estimator(estimator, instance_count, instance_type, da
495495
compression_type (str): Compression type of the input data, if compressed (default: None).
496496
Valid values: 'Gzip', None.
497497
split_type (str): The record delimiter for the input object (default: 'None').
498-
Valid values: 'None', 'Line', and 'RecordIO'.
498+
Valid values: 'None', 'Line', 'RecordIO', and 'TFRecord'.
499499
job_name (str): job name (default: None). If not specified, one will be generated.
500500
strategy (str): The strategy used to decide how to batch records in a single request (default: None).
501501
Valid values: 'MULTI_RECORD' and 'SINGLE_RECORD'.

0 commit comments

Comments
 (0)