diff --git a/saqc/lib/plotting.py b/saqc/lib/plotting.py index 72ae15a72edfe51051241c1203ecfe0de709bb7a..68f59d3a0615e07e63867661add89c0c55bb681a 100644 --- a/saqc/lib/plotting.py +++ b/saqc/lib/plotting.py @@ -3,12 +3,16 @@ import logging import pandas as pd -import numpy as np __plotvars = [] -_colors = dict(unflagged='silver', good='seagreen', bad='firebrick', suspicious='gold') +_colors = { + "unflagged": "silver", + "good": "seagreen", + "bad": "firebrick", + "suspicious": "gold" +} def plotAllHook(data, flagger): @@ -54,7 +58,7 @@ def _plot( else: mpl.use("TkAgg") - if np.isscalar(varname): + if not isinstance(varname, (list, set)): varname = [varname] varname = set(varname)