Format with latest Black version

danielegrattarola · Jun 1, 2023 · 197adfd · 197adfd
1 parent ed62ea1
commit 197adfd
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 4 deletions.
diff --git a/examples/graph_prediction/qm9_ecc.py b/examples/graph_prediction/qm9_ecc.py
@@ -40,6 +40,7 @@
 loader_tr = DisjointLoader(dataset_tr, batch_size=batch_size, epochs=epochs)
 loader_te = DisjointLoader(dataset_te, batch_size=batch_size, epochs=1)
 
+
 ################################################################################
 # Build model
 ################################################################################

diff --git a/examples/other/node_clustering_tvgnn.py b/examples/other/node_clustering_tvgnn.py
@@ -53,6 +53,7 @@
 y = np.argmax(Y, axis=-1)
 n_clust = Y.shape[-1]
 
+
 ################################
 # MODEL
 ################################
@@ -101,6 +102,7 @@ def call(self, inputs):
 model = ClusteringModel(aggr=MP_layers, pool=pool_layer)
 opt = tf.keras.optimizers.Adam(learning_rate=learning_rate)
 
+
 ################################
 # TRAINING
 ################################

diff --git a/spektral/layers/base.py b/spektral/layers/base.py
@@ -118,7 +118,6 @@ def __init__(
         kernel_constraint=None,
         **kwargs
     ):
-
         super().__init__(**kwargs)
         self.trainable_kernel = trainable_kernel
         self.activation = activations.get(activation)
@@ -184,7 +183,6 @@ class MinkowskiProduct(Layer):
     """
 
     def __init__(self, activation=None, **kwargs):
-
         super().__init__(**kwargs)
         self.activation = activations.get(activation)
 

diff --git a/spektral/layers/pooling/dmon_pool.py b/spektral/layers/pooling/dmon_pool.py
@@ -162,7 +162,6 @@ def reduce_index(self, i, s, **kwargs):
         return i_pool
 
     def modularity_loss(self, a, s, a_pool):
-
         if K.is_sparse(a):
             n_edges = tf.cast(len(a.values), dtype=s.dtype)
 

diff --git a/spektral/layers/pooling/global_pool.py b/spektral/layers/pooling/global_pool.py
@@ -8,7 +8,6 @@
 
 class GlobalPool(Layer):
     def __init__(self, **kwargs):
-
         super().__init__(**kwargs)
         self.supports_masking = True
         self.pooling_op = None