refactor: flake8

hnn_core/tests/test_viz.py

@@ -363,7 +363,6 @@ def _get_line_hex_colors(fig):
                                                  colors=dict_mapping)
 
 
-
 def test_network_plotter_init(setup_net):
     """Test init keywords of NetworkPlotter class."""
     net = setup_net

hnn_core/viz.py

@@ -559,19 +559,22 @@ def plot_spikes_raster(cell_response, trial_idx=None, ax=None, show=True,
         default_cell_types = cell_types
 
     # Set default colors
-    default_colors = plt.rcParams['axes.prop_cycle'].by_key()['color'][:len(default_cell_types)]
-    cell_colors = {cell: color for cell, color in zip(default_cell_types, default_colors)}
+    default_colors = (plt.rcParams['axes.prop_cycle']
+                      .by_key()['color'][:len(default_cell_types)])
+    cell_colors = {cell: color
+                   for cell, color in zip(default_cell_types, default_colors)}
 
     # validate colors argument
-    _validate_type(colors, (list, dict, None),'color', 'list of str, or dict')
+    _validate_type(colors, (list, dict, None), 'color', 'list of str, or dict')
     if colors:
         if isinstance(colors, list):
             if len(colors) != len(default_cell_types):
                 raise ValueError(
                     f"Number of colors must be equal to number of "
                     f"cell types. {len(colors)} colors provided "
                     f"for {len(default_cell_types)} cell types.")
-            cell_colors = {cell: color for cell, color in zip(default_cell_types, colors)}
+            cell_colors = {cell: color
+                           for cell, color in zip(default_cell_types, colors)}
 
         if isinstance(colors, dict):
             # Check valid cell types