Skip to content

Commit aae8efa

Browse files
committed
change dd default value
1 parent 24518f5 commit aae8efa

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

src/sagemaker/dataset_definition/inputs.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@ class RedshiftDatasetDefinition(ApiObject):
4949
cluster_role_arn = None
5050
output_s3_uri = None
5151
kms_key_id = None
52-
output_format = "PARQUET"
53-
output_compression = "GZIP"
52+
output_format = None
53+
output_compression = None
5454

5555

5656
class AthenaDatasetDefinition(ApiObject):
@@ -77,8 +77,8 @@ class AthenaDatasetDefinition(ApiObject):
7777
output_s3_uri = None
7878
work_group = None
7979
kms_key_id = None
80-
output_format = "PARQUET"
81-
output_compression = "GZIP"
80+
output_format = None
81+
output_compression = None
8282

8383

8484
class DatasetDefinition(ApiObject):
@@ -102,7 +102,7 @@ class DatasetDefinition(ApiObject):
102102
}
103103

104104
data_distribution_type = "ShardedByS3Key"
105-
input_mode = "Pipe"
105+
input_mode = "File"
106106
local_path = None
107107
redshift_dataset_definition = None
108108
athena_dataset_definition = None

src/sagemaker/processing.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -279,9 +279,7 @@ def _normalize_inputs(self, inputs=None, kms_key=None):
279279
if file_input.input_name is None:
280280
file_input.input_name = "input-{}".format(count)
281281

282-
if isinstance(file_input.source, Properties) or isinstance(
283-
file_input.dataset_definition, DatasetDefinition
284-
):
282+
if isinstance(file_input.source, Properties) or file_input.dataset_definition:
285283
normalized_inputs.append(file_input)
286284
continue
287285

0 commit comments

Comments
 (0)