Included autopep formatting. Will need to be done again after functionality has been ensured

Author: mhehr

analysis/functions/beeswarm.py

@@ -2,7 +2,16 @@
 import matplotlib.pyplot as plt
 import numpy as np
 
-def beeswarm_coordinates(df, val_col, xscale="log", figsize=(9,9), pointsize=7, prefix=''):
+
+def beeswarm_coordinates(
+        df,
+        val_col,
+        xscale="log",
+        figsize=(
+            9,
+            9),
+    pointsize=7,
+        prefix=''):
     '''small hack: don't show image, but use seaborn to calculate coordinates.
     Could not find a solution performing as well as the seaborn implementation'''
     fig, ax = plt.subplots(figsize=figsize)
@@ -11,9 +20,9 @@ def beeswarm_coordinates(df, val_col, xscale="log", figsize=(9,9), pointsize=7,
     # get precise data coordinates
     x, y = np.array(ax.collections[0].get_offsets()).T
     df = df.sort_values(by=val_col, ascending=True)
-    df[prefix+'x'] = x
-    df[prefix+'y'] = y
+    df[prefix + 'x'] = x
+    df[prefix + 'y'] = y
     xlim = ax.get_xlim()
     ylim = ax.get_ylim()
     plt.close('all')
-    return df, xlim, ylim
+    return df, xlim, ylim