Skip to content

Commit e6a734e

Browse files
committed
Update FeatureGenerationTransformer
1 parent 9938b0d commit e6a734e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

hypernets/tests/tabular/feature_generator_dask_test.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -162,17 +162,17 @@ def test_fix_input(self, fix_input: bool):
162162
ftt.fit(df)
163163
x_t = ftt.transform(df)
164164
x_t = x_t.compute()
165-
assert "x1__+__x2" in x_t
166-
assert "x1__/__x2" in x_t
165+
assert "x1__A__x2" in x_t
166+
assert "x1__D__x2" in x_t
167167

168168
if fix_input is True:
169169
# should no NaN value not only input nor output
170170
assert not math.isnan(x_t["x1"][0])
171-
assert not math.isnan(x_t["x1__/__x2"][0])
171+
assert not math.isnan(x_t["x1__D__x2"][0])
172172
else:
173173
# x1 is NaN, it's children is NaN too.
174174
assert math.isnan(x_t["x1"][0])
175-
assert math.isnan(x_t["x1__/__x2"][0])
175+
assert math.isnan(x_t["x1__D__x2"][0])
176176

177177
def test_datetime_derivation(self):
178178

0 commit comments

Comments
 (0)