from deepchem.models.torch_models.torch_model import TorchModel
from deepchem.models.torch_models.attentivefp import AttentiveFP, AttentiveFPModel
from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
from deepchem.models.torch_models.gat import GAT, GATModel
from deepchem.models.torch_models.gcn import GCN, GCNModel
from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
from deepchem.models.torch_models.lcnn import LCNN, LCNNModel
from deepchem.models.torch_models.pagtn import Pagtn, PagtnModel
from deepchem.models.torch_models.mat import MAT, MATModel
from deepchem.models.torch_models.megnet import MEGNetModel
from deepchem.models.torch_models.layers import CNNModule
from deepchem.models.torch_models.cnn import CNN

from deepchem.models.torch_models.torch_model import TorchModel
from deepchem.models.torch_models.attentivefp import AttentiveFP, AttentiveFPModel
from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
from deepchem.models.torch_models.gat import GAT, GATModel
from deepchem.models.torch_models.gcn import GCN, GCNModel
from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
from deepchem.models.torch_models.lcnn import LCNN, LCNNModel
from deepchem.models.torch_models.pagtn import Pagtn, PagtnModel
from deepchem.models.torch_models.mat import MAT, MATModel
from deepchem.models.torch_models.megnet import MEGNetModel
from deepchem.models.torch_models.layers import CNNModule, NeighborList
from deepchem.models.torch_models.cnn import CNN

Add layer to module imports
Add layer to module imports

  from deepchem.models.torch_models.torch_model import TorchModel
  from deepchem.models.torch_models.attentivefp import AttentiveFP, AttentiveFPModel
  from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
  from deepchem.models.torch_models.gat import GAT, GATModel
  from deepchem.models.torch_models.gcn import GCN, GCNModel
  from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
  from deepchem.models.torch_models.lcnn import LCNN, LCNNModel
  from deepchem.models.torch_models.pagtn import Pagtn, PagtnModel
  from deepchem.models.torch_models.mat import MAT, MATModel
  from deepchem.models.torch_models.megnet import MEGNetModel
- from deepchem.models.torch_models.layers import CNNModule
+ from deepchem.models.torch_models.layers import CNNModule, NeighborList
  from deepchem.models.torch_models.cnn import CNN
  
Add layer to module imports
## Code Before:
from deepchem.models.torch_models.torch_model import TorchModel
from deepchem.models.torch_models.attentivefp import AttentiveFP, AttentiveFPModel
from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
from deepchem.models.torch_models.gat import GAT, GATModel
from deepchem.models.torch_models.gcn import GCN, GCNModel
from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
from deepchem.models.torch_models.lcnn import LCNN, LCNNModel
from deepchem.models.torch_models.pagtn import Pagtn, PagtnModel
from deepchem.models.torch_models.mat import MAT, MATModel
from deepchem.models.torch_models.megnet import MEGNetModel
from deepchem.models.torch_models.layers import CNNModule
from deepchem.models.torch_models.cnn import CNN

## Instruction:
Add layer to module imports
## Code After:
from deepchem.models.torch_models.torch_model import TorchModel
from deepchem.models.torch_models.attentivefp import AttentiveFP, AttentiveFPModel
from deepchem.models.torch_models.cgcnn import CGCNN, CGCNNModel
from deepchem.models.torch_models.gat import GAT, GATModel
from deepchem.models.torch_models.gcn import GCN, GCNModel
from deepchem.models.torch_models.mpnn import MPNN, MPNNModel
from deepchem.models.torch_models.lcnn import LCNN, LCNNModel
from deepchem.models.torch_models.pagtn import Pagtn, PagtnModel
from deepchem.models.torch_models.mat import MAT, MATModel
from deepchem.models.torch_models.megnet import MEGNetModel
from deepchem.models.torch_models.layers import CNNModule, NeighborList
from deepchem.models.torch_models.cnn import CNN
