fix a bug in conftest.py

bigdata-ustc · Apr 7, 2021 · 65d8d44 · 65d8d44
1 parent 1d9ad8b
commit 65d8d44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ncdm/conftest.py b/tests/ncdm/conftest.py
@@ -35,7 +35,7 @@ def data(conf):
     dataset = TensorDataset(
         torch.tensor(user_id, dtype=torch.int64),
         torch.tensor(item_id, dtype=torch.int64),
-        torch.tensor(knowledge_embs, dtype=torch.int64),
+        torch.tensor(knowledge_emb, dtype=torch.int64),
         torch.tensor(score, dtype=torch.float)
     )
     return DataLoader(dataset, batch_size=batch_size)