Lintering

- set ignore RET504 rule for ruff
- some lintering

custom_models/cell/cin.py

@@ -3,6 +3,9 @@
 import torch
 import torch.nn.functional as F
 from topomodelx.nn.cell.cwn_layer import CWNLayer
+import torch.nn as nn
+from topomodelx.base.conv import Conv
+from torch_geometric.nn.models import MLP
 
 
 class CWN(torch.nn.Module):
@@ -109,11 +112,6 @@ def forward(
 
 """Implementation of CWN layer from Bodnar et al.: Weisfeiler and Lehman Go Cellular: CW Networks."""
 
-import torch.nn as nn
-import torch.nn.functional as F
-from topomodelx.base.conv import Conv
-from torch_geometric.nn.models import MLP
-
 
 class CWNLayer(nn.Module):
     r"""Layer of a CW Network (CWN).

custom_models/cell/cwn_dcm.py

@@ -5,7 +5,7 @@
 
 class MLP(nn.Module):
     def __init__(self, layers_size, dropout=0.0, final_activation=False):
-        super(MLP, self).__init__()
+        super().__init__()
         layers = []
         for li in range(1, len(layers_size)):
             layers.append(nn.Dropout(p=dropout))
@@ -22,7 +22,7 @@ def forward(self, x, e=None):
 
 class CW(nn.Module):
     def __init__(self, F_in, F_out):
-        super(CW, self).__init__()
+        super().__init__()
         self.har = nn.Linear(F_in, F_out)
         self.sol = GCNConv(F_in, F_out, add_self_loops=False)
         self.irr = GCNConv(F_in, F_out, add_self_loops=False)
@@ -36,7 +36,7 @@ def forward(self, xe, Lu, Ld):
 
 class CWNDCM(nn.Module):
     def __init__(self, in_channels, n_layers=2, dropout=0.0, last_act=False):
-        super(CWNDCM, self).__init__()
+        super().__init__()
         self.d = dropout
         self.convs = nn.ModuleList()
         self.last_act = last_act

pyproject.toml

@@ -101,7 +101,10 @@ select = [
     "PERF", # performance rules
     "RUF", # miscellaneous rules
 ]
-ignore = ["E501"] # line too long
+ignore = [
+    "E501", # line too long
+    "RET504" # Unnecessary assignment before return
+] 
 
 [tool.ruff.lint.pydocstyle]
 convention = "numpy"