adjusted dependencies for Windows 10

Author: SaizhuoWang

.gitignore

@@ -1,3 +1,4 @@
+config
 # IDE
 .idea
 run_*.sh

configure

@@ -4,4 +4,4 @@ read -p '[1] for CUDA 9.2, [2] for CUDA 10.1, [3] for CUDA 10.2, [4] for CUDA 11
 echo $CUDA_version > config
 
 echo 'Configuration completed.'
-echo 'Run `pip install .` or `python setup.py install` to install graph4nlp in your environment.'
+echo 'Run `python setup.py install` to install graph4nlp in your environment.'

configure.bat

@@ -3,4 +3,4 @@ echo Please specify your CUDA version
 set /p cuda_version="[1] for CUDA 9.2, [2] for CUDA 10.1, [3] for CUDA 10.2, [4] for CUDA 11.0, [5] for CUDA 11.1, [6] for CPU: "
 echo %cuda_version% > config
 echo Configuration completed.
-echo Run `pip install .` to install graph4nlp in your environment.
+echo Run `python setup.py install` to install graph4nlp in your environment.

graph4nlp/pytorch/data/data.py

@@ -570,14 +570,14 @@ def to_dgl(self) -> dgl.DGLGraph:
         g: dgl.DGLGraph
             The converted dgl.DGLGraph
         """
-        dgl_g = dgl.DGLGraph().to(self.device)
+        u, v = self._edge_indices.src, self._edge_indices.tgt
+        num_nodes = self.get_node_num()
+        dgl_g = dgl.graph(u, v, num_nodes=num_nodes).to(self.device)
         # Add nodes and their features
-        dgl_g.add_nodes(num=self.get_node_num())
         for key, value in self._node_features.items():
             if value is not None:
                 dgl_g.ndata[key] = value
         # Add edges and their features
-        dgl_g.add_edges(u=self._edge_indices.src, v=self._edge_indices.tgt)
         for key, value in self._edge_features.items():
             if value is not None:
                 dgl_g.edata[key] = value

setup.py

@@ -37,7 +37,10 @@ def finalize_options(self):
     cuda_version = cuda_versions[version]
 
     if cuda_version == '11.1':
-        os.system("pip3 install torch==1.8.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html")
+        os.system("pip install torch==1.8.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html")
+
+    if platform.system() == 'Windows':
+        os.system('pip install torch==1.8.1+cu102 -f https://download.pytorch.org/whl/torch_stable.html')
 
     if cuda_version == 'none':
         cuda_version = ''
@@ -48,7 +51,7 @@ def finalize_options(self):
 
     install_requirement = ['pythonds', 'nltk >= 3.5', 'stanfordcorenlp', 'scipy >= 1.5.2',
                            'scikit-learn >= 0.23.2', 'networkx >= 2.5', 'dgl{} >= 0.4'.format(cuda_version),
-                           'ogb', 'tqdm >= 4.29.0', 'pyyaml']
+                           'ogb', 'torchtext', 'tqdm >= 4.29.0', 'pyyaml', 'transformers']
     pytorch_requirement = 'torch >= 1.6.0' if platform.system() != 'Windows' else 'torch >= 1.8.1'
     torchtext_requirement = 'torchtext == 0.7.0' if platform.system() != 'Windows' else 'torchtext == 0.9.0'
     install_requirement.append(pytorch_requirement)