diff --git a/kwant/plotter.py b/kwant/plotter.py index 694f84d9efdfe516783ff28cd35528e581c6a424..88cb7256ea1b7b39b8636020a7c96820229e0307 100644 --- a/kwant/plotter.py +++ b/kwant/plotter.py @@ -1275,12 +1275,8 @@ def plot(sys, num_lead_cells=2, unit='nn', hop_cmap = None if not isinstance(cmap, basestring): try: - if len(cmap) != 2: - raise ValueError("if cmap is a sequence, it must have exactly " - "two entries") - hop_cmap = cmap[1] - cmap = cmap[0] - except (TypeError, KeyError): + cmap, hop_cmap = cmap + except TypeError: pass # make a new figure unless axes specified