We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3d6eba3 commit 267a424Copy full SHA for 267a424
tests/prediction_pipeline/test_preprocessing.py
@@ -159,8 +159,8 @@ def test_sigmoid():
159
np_data = np.random.rand(*shape)
160
data = xr.DataArray(np_data, dims=axes)
161
162
- sigmoid = make_preprocessing([Preprocessing("sigmoid")])
+ sigmoid = make_preprocessing([Preprocessing("sigmoid", kwargs={})])
163
res = sigmoid(data)
164
165
- exp = 1.0 / (1 + np.exp(-np_data))
+ exp = xr.DataArray(1.0 / (1 + np.exp(-np_data)), dims=axes)
166
xr.testing.assert_allclose(res, exp)
0 commit comments