more linting

Author: mcgalcode

src/pylattica/visualization/result_artist.py

@@ -97,7 +97,10 @@ def jupyter_play(self, cell_size: int = 20, wait: int = 1, **kwargs):
         wait : int, optional
             The time duration between frames in the animation. Defaults to 1., by default 1
         """
-        from IPython.display import clear_output, display  # pylint: disable=import-error  # pragma: no cover
+        from IPython.display import (
+            clear_output,
+            display,
+        )  # pylint: disable=import-error  # pragma: no cover
 
         imgs = self._get_images(cell_size=cell_size, **kwargs)  # pragma: no cover
         for img in imgs:  # pragma: no cover

src/pylattica/visualization/structure_artist.py

@@ -29,7 +29,9 @@ def jupyter_show(self, state: SimulationState, **kwargs):
         state : SimulationState
             The simulation state to display.
         """
-        from IPython.display import display  # pylint: disable=import-error  # pragma: no cover
+        from IPython.display import (
+            display,
+        )  # pylint: disable=import-error  # pragma: no cover
 
         img = self.get_img(state, **kwargs)  # pragma: no cover
         display(img)  # pragma: no cover