finish fixing HPF tests

lenskit · Dec 24, 2024 · a3e3b3a · a3e3b3a
1 parent 79ad05f
commit a3e3b3a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lenskit-hpf/tests/test_hpf.py b/lenskit-hpf/tests/test_hpf.py
@@ -44,7 +44,7 @@ def test_hpf_train_large(tmp_path, ml_ratings):
     pipe = topn_pipeline(algo)
     pipe.train(ds, retrain=False)
 
-    for u in np.random.choice(ratings.user.unique(), size=50, replace=False):
+    for u in np.random.choice(ratings.user_id.unique(), size=50, replace=False):
         recs = pipe.run("recommender", query=u, n=50)
         assert isinstance(recs, ItemList)
         assert len(recs) == 50
@@ -73,7 +73,7 @@ def test_hpf_train_binary(tmp_path, ml_ratings):
     pipe = topn_pipeline(algo)
     pipe.train(ds, retrain=False)
 
-    for u in np.random.choice(ratings.user.unique(), size=50, replace=False):
+    for u in np.random.choice(ratings.user_id.unique(), size=50, replace=False):
         recs = pipe.run("recommender", query=u, n=50)
         assert isinstance(recs, ItemList)
         assert len(recs) == 50