linter

Author: GiovanniCanali

pina/model/block/message_passing/interaction_network_block.py

@@ -31,6 +31,7 @@ class InteractionNetworkBlock(MessagePassing):
         In Advances in Neural Information Processing Systems (NeurIPS 2016).
         DOI: `<https://doi.org/10.48550/arXiv.1612.00222>_`.
     """
+
     def __init__(
         self,
         node_feature_dim,
@@ -84,19 +85,18 @@ def __init__(
                 "`node_feature_dim` must be a positive integer,"
                 f" got {node_feature_dim}."
             )
-        
+
         if hidden_dim <= 0:
             raise ValueError(
-                "`hidden_dim` must be a positive integer,"
-                f" got {hidden_dim}."
+                "`hidden_dim` must be a positive integer," f" got {hidden_dim}."
             )
-        
+
         if n_message_layers <= 0:
             raise ValueError(
                 "`n_message_layers` must be a positive integer,"
                 f" got {n_message_layers}."
             )
-        
+
         if n_update_layers <= 0:
             raise ValueError(
                 "`n_update_layers` must be a positive integer,"
@@ -110,7 +110,7 @@ def __init__(
 
         # Message network
         self.message_net = FeedForward(
-            input_dimensions=2*self.node_feature_dim,
+            input_dimensions=2 * self.node_feature_dim,
             output_dimensions=self.hidden_dim,
             inner_size=self.hidden_dim,
             n_layers=n_message_layers,
@@ -165,4 +165,4 @@ def update(self, message, x):
         :return: The updated node features.
         :rtype: torch.Tensor
         """
-        return self.update_net(torch.cat((x, message), dim=-1))
+        return self.update_net(torch.cat((x, message), dim=-1))