lie_conv/graphnets.py
Function __init__
has 6 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def __init__(self,num_species,charge_scale,num_outputs=1,d=3,k=64,num_layers=1):
Function forward
has 5 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def forward(self, x, edge_index, edge_attr, u, batch):
Function forward
has 5 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def forward(self, x, edge_index, edge_attr, u, batch):
Function forward
has 5 arguments (exceeds 4 allowed). Consider refactoring. Open
Open
def forward(self, src, dest, edge_attr, u, batch):