From 5514ffd16c9bb0dd09e5b78c762c01fe083cb76b Mon Sep 17 00:00:00 2001 From: AnotherSamWilson Date: Sun, 28 Jul 2024 10:53:02 -0400 Subject: [PATCH] Fixed reference to renamed method. --- tests/test_imputed_accuracy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_imputed_accuracy.py b/tests/test_imputed_accuracy.py index 53ab826..bbbd157 100644 --- a/tests/test_imputed_accuracy.py +++ b/tests/test_imputed_accuracy.py @@ -78,7 +78,7 @@ def get_categorical_performance(kernel: mf.ImputationKernel, variables, iris): cand = kernel._make_label(col, seed=model.params["seed"]) orig = iris.loc[ind, col] imps = kernel[col, iterations, 0] - bf = kernel.get_bachelor_features(col) + bf = kernel._get_bachelor_features(col) preds = model.predict(bf) rocs[col] = roc_auc_score(orig, preds, multi_class="ovr", average="macro") accs[col] = (imps == orig).mean()