use same metric for rct and rot

apmellot · Dec 7, 2023 · 28092d0 · 28092d0
1 parent 0d8a8e9
commit 28092d0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_transfer.py b/tests/test_transfer.py
@@ -135,8 +135,8 @@ def test_tlrotate_fit_transform(rndstate, metric, sample_weight):
         sample_weight_ = None
     sample_weight_ = check_weights(sample_weight_, len(y_enc))
 
-    rct = TLCenter(target_domain='target_domain')
-    X_rct = rct.fit_transform(X, y_enc, sample_weight_)
+    rct = TLCenter(target_domain='target_domain', metric=metric)
+    X_rct = rct.fit_transform(X, y_enc, sample_weight=sample_weight_)
     rot = TLRotate(target_domain='target_domain', metric=metric)
     X_rot = rot.fit_transform(X_rct, y_enc, sample_weight=sample_weight_)