Rename classes and modules for GNO

pina/model/layers/__init__.py

@@ -15,7 +15,7 @@ __all__ = [
     "AVNOBlock",
     "LowRankBlock",
     "RBFBlock",
-    "GraphIntegralLayer"
+    "GNOBlock"
 ]
 
 from .convolution_2d import ContinuousConvBlock
@@ -32,4 +32,4 @@ from .embedding import PeriodicBoundaryEmbedding, FourierFeatureEmbedding
 from .avno_layer import AVNOBlock
 from .lowrank_layer import LowRankBlock
 from .rbf_layer import RBFBlock
-from .graph_integral_kernel import GraphIntegralLayer
+from .gno_block import GNOBlock

pina/model/layers/gno_block.py

@@ -2,10 +2,11 @@ import torch
 from torch_geometric.nn import MessagePassing
 
 
-class GraphIntegralLayer(MessagePassing):
+class GNOBlock(MessagePassing):
     """
     TODO: Add documentation
     """
+
     def __init__(
             self,
             width,
@@ -27,7 +28,7 @@ class GraphIntegralLayer(MessagePassing):
         :type n_layers: int
         """
         from pina.model import FeedForward
-        super(GraphIntegralLayer, self).__init__(aggr='mean')
+        super(GNOBlock, self).__init__(aggr='mean')
         self.width = width
         if layers is None and inner_size is None:
             inner_size = width