Skip to content

Commit 121a247

Browse files
author
William de Vazelhes
committed
FIX: fix error message for SDML
1 parent e8c74d0 commit 121a247

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

metric_learn/sdml.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def _fit(self, pairs, y):
8888
"positive semi-definite (PSD). The algorithm may diverge, "
8989
"and lead to degenerate solutions. "
9090
"To prevent that, try to decrease the balance parameter "
91-
"`balance_param` and/or to set use_covariance=False.",
91+
"`balance_param` and/or to set use_cov=False.",
9292
ConvergenceWarning)
9393
w -= min_eigval # we translate the eigenvalues to make them all positive
9494
w += 1e-10 # we add a small offset to avoid definiteness problems

test/metric_learn_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,7 @@ def test_sdml_raises_warning_non_psd(self):
295295
"positive semi-definite (PSD). The algorithm may diverge, "
296296
"and lead to degenerate solutions. "
297297
"To prevent that, try to decrease the balance parameter "
298-
"`balance_param` and/or to set use_covariance=False.")
298+
"`balance_param` and/or to set use_cov=False.")
299299
with pytest.warns(ConvergenceWarning) as raised_warning:
300300
try:
301301
sdml.fit(pairs, y)

0 commit comments

Comments
 (0)