black check

Author: Raphael Meudec

README.md

@@ -288,3 +288,7 @@ model.fit(x_train, y_train, batch_size=2, epochs=2, callbacks=callbacks)
 ## Contributing
 
 To contribute to the project, please read the [dedicated section](./CONTRIBUTING.md).
+
+## Citation
+
+A [citation file](./CITATION.cff) is available for citing this work.

tf_explain/callbacks/activations_visualization.py

@@ -13,7 +13,7 @@
 
 class ActivationsVisualizationCallback(Callback):
 
-    """ Draw activations of a specific layer for a given input """
+    """Draw activations of a specific layer for a given input"""
 
     def __init__(
         self,

tf_explain/core/activations.py

@@ -10,7 +10,7 @@
 
 class ExtractActivations:
 
-    """ Draw activations of a specific layer for a given input """
+    """Draw activations of a specific layer for a given input"""
 
     def __init__(self, batch_size=None):
         self.batch_size = batch_size