From 632bcbba405c98bb270a547c89f7622f331ad097 Mon Sep 17 00:00:00 2001
From: Ben Krikler <bek07@ic.ac.uk>
Date: Tue, 26 Mar 2019 13:28:41 +0100
Subject: [PATCH] Fix lint
---
fast_plotter/plotting.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/fast_plotter/plotting.py b/fast_plotter/plotting.py
index f445b9d..6e316f4 100644
--- a/fast_plotter/plotting.py
+++ b/fast_plotter/plotting.py
@@ -49,7 +49,7 @@ def plot_all(df, project_1d=True, project_2d=True, data="data", signal=None, dat
return figures, ran_ok
-class fill_coll(object):
+class FillColl(object):
def __init__(self, n_colors=10, ax=None, fill=True, line=True):
self.calls = 0
colormap = plt.cm.nipy_spectral
@@ -88,7 +88,7 @@ class fill_coll(object):
self.calls += 1
-class bar_coll(fill_coll):
+class BarColl(FillColl):
def __call__(self, col, **kwargs):
ax, x, y, color = self.pre_call(col)
align = "center"
@@ -107,14 +107,14 @@ def actually_plot(df, x_axis, y, yerr, kind, label, ax, dataset_col="dataset"):
return
n_datasets = len(df.index.unique(dataset_col))
if kind == "line":
- filler = fill_coll(n_datasets, ax=ax, fill=False)
+ filler = FillColl(n_datasets, ax=ax, fill=False)
df[y].unstack(dataset_col).iloc[:, ::-1].apply(filler, axis=0, step="mid")
return
elif kind == "bar":
- filler = bar_coll(n_datasets, ax=ax)
+ filler = BarColl(n_datasets, ax=ax)
df[y].unstack(dataset_col).iloc[:, ::-1].apply(filler, axis=0, step="mid")
elif kind == "fill":
- filler = fill_coll(n_datasets, ax=ax)
+ filler = FillColl(n_datasets, ax=ax)
df[y].unstack(dataset_col).iloc[:, ::-1].apply(filler, axis=0, step="mid")
elif kind == "fill-error-last":
actually_plot(df, x_axis, y, yerr, "fill",
--
GitLab