Skip to content

Commit 267a424

Browse files
Fix sigmoid test
1 parent 3d6eba3 commit 267a424

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/prediction_pipeline/test_preprocessing.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,8 +159,8 @@ def test_sigmoid():
159159
np_data = np.random.rand(*shape)
160160
data = xr.DataArray(np_data, dims=axes)
161161

162-
sigmoid = make_preprocessing([Preprocessing("sigmoid")])
162+
sigmoid = make_preprocessing([Preprocessing("sigmoid", kwargs={})])
163163
res = sigmoid(data)
164164

165-
exp = 1.0 / (1 + np.exp(-np_data))
165+
exp = xr.DataArray(1.0 / (1 + np.exp(-np_data)), dims=axes)
166166
xr.testing.assert_allclose(res, exp)

0 commit comments

Comments
 (0)