Skip to content

Commit c96e4ef

Browse files
Update test FE
1 parent 0169094 commit c96e4ef

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_FE.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,6 @@ def predict(self, X):
1717
def test_fit_predict():
1818
model = FE(DummyEstimator)
1919
model.fit(np.ones((100,10)), np.ones(100),
20-
index=[range(80), range(80, 100)])
20+
src_index=range(80), tgt_index=range(80, 100))
2121
y_pred = model.predict(np.ones((100,10)))
2222
assert np.all(y_pred == np.zeros(100))

0 commit comments

Comments
 (0)