fix stray test failures

lenskit · Jan 21, 2025 · 4a8d383 · 4a8d383
1 parent 56504c6
commit 4a8d383
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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]))