@@ -163,9 +163,15 @@ def test_with_feature_group_with_additional_params(sagemaker_session_mock):
163
163
{"FeatureName" : "feature-2" , "FeatureType" : "String" },
164
164
],
165
165
}
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
+
169
175
assert len (dataset_builder ._feature_groups_to_be_merged ) == 1
170
176
assert dataset_builder ._feature_groups_to_be_merged [0 ].features == [
171
177
"feature-1" ,
@@ -189,10 +195,7 @@ def test_with_feature_group_with_additional_params(sagemaker_session_mock):
189
195
dataset_builder ._feature_groups_to_be_merged [0 ].event_time_identifier_feature .feature_type
190
196
== FeatureTypeEnum .STRING
191
197
)
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
196
199
assert (
197
200
dataset_builder ._feature_groups_to_be_merged [0 ].join_comparator
198
201
== JoinComparatorEnum .LESS_THAN
0 commit comments