From 4a8d3838dedfd3fb719e1ef0c15ede24f85ffad4 Mon Sep 17 00:00:00 2001 From: Michael Ekstrand Date: Tue, 21 Jan 2025 02:11:28 -0500 Subject: [PATCH] fix stray test failures --- lenskit/tests/models/test_als_explicit.py | 2 +- lenskit/tests/models/test_knn_item_item.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lenskit/tests/models/test_als_explicit.py b/lenskit/tests/models/test_als_explicit.py index 213bcf03c..94ef31531 100644 --- a/lenskit/tests/models/test_als_explicit.py +++ b/lenskit/tests/models/test_als_explicit.py @@ -217,7 +217,7 @@ def test_als_train_large(ml_ratings, ml_ds: Dataset): assert len(algo.items_) == ml_ratings["item_id"].nunique() assert len(algo.users_) == ml_ratings["user_id"].nunique() - ratings = ml_ds.interaction_matrix(format="pandas", field="all") + ratings = ml_ds.interaction_matrix(format="pandas") gmean = ratings["rating"].mean() istats = ml_ds.item_stats() diff --git a/lenskit/tests/models/test_knn_item_item.py b/lenskit/tests/models/test_knn_item_item.py index f9a2f07d9..06147d9cf 100644 --- a/lenskit/tests/models/test_knn_item_item.py +++ b/lenskit/tests/models/test_knn_item_item.py @@ -157,7 +157,7 @@ def test_ii_simple_implicit_predict(): history = UserTrainingHistoryLookup() history.train(simple_ds) algo = ItemKNNScorer(k=30, feedback="implicit") - algo.train(from_interactions_df(simple_ratings.loc[:, ["user", "item"]])) + algo.train(from_interactions_df(simple_ratings.loc[:, ["user_id", "item_id"]])) q = history(3) res = algo(q, ItemList([6]))