Add pytorch_geometric
@@ -20,11 +20,11 @@ requirements:
- python >=3.7
- setuptools >=38.6.0
run:
- # - argparse >=1.4.0
- numpy >=1.19.4
- networkx
- python >=3.7
- - pytorch ==1.10.0
+ - pytorch >=1.10.0
+ - pytorch_geometric >=2.0.3
- scikit-learn >=0.22.1
- scipy >=1.5.2
- setuptools >=50.3.1.post20201107
@@ -32,6 +32,9 @@ requirements:
test:
imports:
- pygod
+ - pygod.models
+ - pygod.test
+ - pygod.utils
commands:
- pip check
requires:
