Skip to content

Commit 56fc682

Browse files
runs lint for test_dataset_builder
1 parent f34ac66 commit 56fc682

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

tests/unit/sagemaker/feature_store/test_dataset_builder.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,15 @@ def test_with_feature_group_with_additional_params(sagemaker_session_mock):
163163
{"FeatureName": "feature-2", "FeatureType": "String"},
164164
],
165165
}
166-
dataset_builder.with_feature_group(feature_group, "target-feature", [
167-
"feature-1", "feature-2"], join_comparator=JoinComparatorEnum.LESS_THAN, join_type=JoinTypeEnum.LEFT_JOIN, feature_name_in_target='feature-2')
168-
166+
dataset_builder.with_feature_group(
167+
feature_group,
168+
"target-feature",
169+
["feature-1", "feature-2"],
170+
join_comparator=JoinComparatorEnum.LESS_THAN,
171+
join_type=JoinTypeEnum.LEFT_JOIN,
172+
feature_name_in_target="feature-2",
173+
)
174+
169175
assert len(dataset_builder._feature_groups_to_be_merged) == 1
170176
assert dataset_builder._feature_groups_to_be_merged[0].features == [
171177
"feature-1",
@@ -189,10 +195,7 @@ def test_with_feature_group_with_additional_params(sagemaker_session_mock):
189195
dataset_builder._feature_groups_to_be_merged[0].event_time_identifier_feature.feature_type
190196
== FeatureTypeEnum.STRING
191197
)
192-
assert (
193-
dataset_builder._feature_groups_to_be_merged[0].join_type
194-
== JoinTypeEnum.LEFT_JOIN
195-
)
198+
assert dataset_builder._feature_groups_to_be_merged[0].join_type == JoinTypeEnum.LEFT_JOIN
196199
assert (
197200
dataset_builder._feature_groups_to_be_merged[0].join_comparator
198201
== JoinComparatorEnum.LESS_THAN

0 commit comments

Comments
 (0)