fix flake8

pytorch · Jan 3, 2023 · 98f1efc · 98f1efc
1 parent 7c16b42
commit 98f1efc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/captum/attr/_core/latent_shift.py b/captum/attr/_core/latent_shift.py
@@ -264,7 +264,7 @@ def generate_video(
             verbose: True to print debug text
             extra_loops: The video does one loop by default. This will repeat
                 those loops to make it easier to watch.
-            cmap: The cmap value passed to matplotlib. e.g. 'gray' for a 
+            cmap: The cmap value passed to matplotlib. e.g. 'gray' for a
                 grayscale image.
 
         Returns:
@@ -283,7 +283,7 @@ def generate_video(
         # Add reversed so we have an animation cycle
         towrite = list(reversed(imgs)) + list(imgs)
         ys = list(reversed(params["preds"])) + list(params["preds"])
-        
+
         for n in range(extra_loops):
             towrite += towrite
             ys += ys