Skip to content

Commit 782cf46

Browse files
author
mvargas33
committed
Fixed more test regarding rename of variable
1 parent 01c3e9c commit 782cf46

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/test_base_metric.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,12 @@ def test_covariance(self):
3636
remove_spaces(f"Covariance({merged_kwargs})"))
3737

3838
def test_lmnn(self):
39-
def_kwargs = {'convergence_tol': 0.001, 'init': 'auto', 'k': 3,
39+
def_kwargs = {'convergence_tol': 0.001, 'init': 'auto', 'n_neighbors': 3,
4040
'learn_rate': 1e-07, 'max_iter': 1000, 'min_iter': 50,
4141
'n_components': None, 'preprocessor': None,
4242
'random_state': None, 'regularization': 0.5,
4343
'verbose': False}
44-
nndef_kwargs = {'convergence_tol': 0.01, 'k': 6}
44+
nndef_kwargs = {'convergence_tol': 0.01, 'n_neighbors': 6}
4545
merged_kwargs = sk_repr_kwargs(def_kwargs, nndef_kwargs)
4646
self.assertEqual(
4747
remove_spaces(str(metric_learn.LMNN(convergence_tol=0.01, n_neighbors=6))),

test/test_mahalanobis_mixin.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,7 @@ def test_auto_init_transformation(n_samples, n_features, n_classes,
417417
random_state=rng)
418418
# To make the test work for LMNN:
419419
if 'LMNN' in model_base.__class__.__name__:
420-
model_base.set_params(k=1)
420+
model_base.set_params(n_neighbors=1)
421421
# To make the test faster for estimators that have a max_iter:
422422
if hasattr(model_base, 'max_iter'):
423423
model_base.set_params(max_iter=1)

0 commit comments

Comments
 (0)