Merge pull request #61 from Fung-Lab/torchmd_fix_scatter

Use torch_geometric.utils for scatter

matdeeplearn/models/torchmd_etEarly.py

@@ -4,7 +4,7 @@
 from torch import Tensor, nn
 import torch_geometric.nn
 from torch_geometric.nn import MessagePassing
-from torch_scatter import scatter
+from torch_geometric.utils import scatter
 from matdeeplearn.models.utils import (
     NeighborEmbedding,
     CosineCutoff,