linter

Author: GiovanniCanali

pina/model/block/message_passing/radial_field_network_block.py

@@ -37,7 +37,7 @@ class RadialFieldNetworkBlock(MessagePassing):
     def __init__(
         self,
         node_feature_dim,
-        hidden_dim = 64,
+        hidden_dim=64,
         n_layers=2,
         activation=torch.nn.Tanh,
         aggr="add",

pina/model/block/message_passing/schnet_block.py

@@ -37,7 +37,7 @@ class SchnetBlock(MessagePassing):
     def __init__(
         self,
         node_feature_dim,
-        hidden_dim = 64,
+        hidden_dim=64,
         n_message_layers=2,
         n_update_layers=2,
         n_radial_layers=2,
@@ -107,7 +107,7 @@ def __init__(
 
         # Layer for updating the node features
         self.update_net = FeedForward(
-            input_dimensions=2*node_feature_dim,
+            input_dimensions=2 * node_feature_dim,
             output_dimensions=node_feature_dim,
             inner_size=hidden_dim,
             n_layers=n_update_layers,