fix installation configuration (#223)

* remove the test of modelnet40 dataset

* fix installation configuration

README.md

@@ -142,7 +142,7 @@ Currently, GammaGL requires **Python Version >= 3.9** and is only supported on *
 
    # For torch, version 2.1+cuda 11.8
    # https://pytorch.org/get-started/locally/
-   $ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
+   $ pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
 
    # For paddle, any latest stable version
    # https://www.paddlepaddle.org.cn/

setup.py

@@ -118,8 +118,8 @@ def load_extensions():
 
     return extensions
 
-install_requires = ['numpy==1.24', 'pandas', 'numba', 'scipy', 'protobuf', 'pyparsing', 'rdkit',
-                    'tensorboardx', 'pytest', 'tensorlayerx', 'rich', 'tqdm', 'pybind11', 'panda', 'ninja']
+install_requires = ['numpy==1.24', 'pandas', 'numba==0.59.0', 'scipy', 'protobuf', 'pyparsing', 'rdkit',
+                    'tensorboardx', 'pytest', 'tensorlayerx', 'rich', 'tqdm', 'pybind11', 'panda', 'ninja==1.11.1.1']
 
 classifiers = [
     'Development Status :: 3 - Alpha',

tests/datasets/test_modelnet40.py

@@ -1,14 +1,14 @@
-from gammagl.datasets import ModelNet40
+# from gammagl.datasets import ModelNet40
 
-root = './data'
-def test_modelnet40(get_dataset):
-    train_dataset = get_dataset(name = 'ModelNet40')
-    assert train_dataset.num_features == 3
-    assert train_dataset.num_node_features == 3
-    assert train_dataset.num_points == 1024
-    assert train_dataset.split == 'train'
-    test_dataset = get_dataset(name = 'ModelNet40',split = 'test')
-    assert test_dataset.num_features == 3
-    assert test_dataset.num_node_features == 3
-    assert test_dataset.num_points == 1024
-    assert test_dataset.split == 'test'
+# root = './data'
+# def test_modelnet40(get_dataset):
+#     train_dataset = get_dataset(name = 'ModelNet40')
+#     assert train_dataset.num_features == 3
+#     assert train_dataset.num_node_features == 3
+#     assert train_dataset.num_points == 1024
+#     assert train_dataset.split == 'train'
+#     test_dataset = get_dataset(name = 'ModelNet40',split = 'test')
+#     assert test_dataset.num_features == 3
+#     assert test_dataset.num_node_features == 3
+#     assert test_dataset.num_points == 1024
+#     assert test_dataset.split == 'test'