pre-commit fixes

mpl-extensions · Oct 27, 2021 · fdc3dfc · fdc3dfc
1 parent da46f0e
commit fdc3dfc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/lazy_loading.py b/examples/lazy_loading.py
@@ -2,11 +2,11 @@
 # if you do this then you must also provide *N_images* in the labeller constructor
 
 
-from mpl_image_labeller import image_labeller
 import matplotlib.pyplot as plt
-
 from numpy.random import default_rng
 
+from mpl_image_labeller import image_labeller
+
 
 def lazy_image_generator(idx):
     rng = default_rng(idx)

diff --git a/mpl_image_labeller/_labeller.py b/mpl_image_labeller/_labeller.py
@@ -57,10 +57,13 @@ def __init__(
                     "If images is a callable then N_images must be provided"
                 )
             self._N_images = N_images
+
             def _get_image(i):
                 return self._images(i)
+
         else:
             self._N_images = len(images)
+
             def _get_image(i):
                 return self._images[i]