Linting

benedekrozemberczki · Jul 2, 2020 · 2806eaf · 2806eaf
1 parent 9e8f08e
commit 2806eaf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric_temporal/nn/conv/gconv_gru.py b/torch_geometric_temporal/nn/conv/gconv_gru.py
@@ -18,7 +18,6 @@ def __init__(self, in_channels, out_channels, K):
         self.in_channels = in_channels
         self.out_channels = out_channels
         self.K = K
-
         self._create_parameters_and_layers()
 
 
@@ -66,6 +65,7 @@ def _set_hidden_state(self, X, H):
             H = torch.zeros(X.shape[0], self.out_channels)
         return H
 
+
     def _calculate_update_gate(self, X, edge_index, edge_weight, H):
         Z = self.conv_x_z(X, edge_index, edge_weight)
         Z = Z + self.conv_h_z(H, edge_index, edge_weight)