fixed: pytest

easy-graph · Jan 18, 2025 · ca73fef · ca73fef
1 parent 564770d
commit ca73fef
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -206,3 +206,4 @@ Temporary Items
 .apdisk
 tmp_test/allset_test.py
 tmp_test/
+RandomNetwork.txt
diff --git a/easygraph/nn/tests/test_regularization.py b/easygraph/nn/tests/test_regularization.py
@@ -2,10 +2,12 @@
 import pytest
 import torch
 
+from easygraph.nn.regularization import EmbeddingRegularization
+
 
 def test_embedding_reg():
     print("EmbeddingRegularization" in eg.__dir__())
-    emb_reg = eg.EmbeddingRegularization(p=2, weight_decay=1e-4)
+    emb_reg = EmbeddingRegularization(p=2, weight_decay=1e-4)
     embs = [torch.randn(10, 3), torch.randn(10, 3)]
     loss = emb_reg(*embs)
     true_loss = 0

diff --git a/requirements.txt b/requirements.txt
@@ -8,5 +8,8 @@ pandas>=2.0.1
 nose>=1.3.7
 pybind11>=2.10.4
 pydsge
+torch-geometric
+torch-sparse
+torch-scatter
 torch >= 1.12.1
-requests
+requests