Skip to content

Commit 81901ee

Browse files
Fix mahalanobis test
1 parent 7102bc5 commit 81901ee

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/test_mahalanobis_mixin.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -215,8 +215,7 @@ def test_get_metric_equivalent_to_explicit_mahalanobis(estimator,
215215
metric = model.get_metric()
216216
n_features = X.shape[1]
217217
a, b = (rng.randn(n_features), rng.randn(n_features))
218-
expected_dist = mahalanobis(a[None], b[None],
219-
VI=model.get_mahalanobis_matrix())
218+
expected_dist = mahalanobis(a, b, VI=model.get_mahalanobis_matrix())
220219
assert_allclose(metric(a, b), expected_dist, rtol=1e-13)
221220

222221

0 commit comments

Comments
 (0)