From 40523683a6fc5fe71cc6de4344a6a607f200b963 Mon Sep 17 00:00:00 2001
From: Michael Wimmer <wimmer@lorentz.leidenuniv.nl>
Date: Tue, 27 Aug 2013 14:28:43 +0200
Subject: [PATCH] fix bug that showed up when specifying color as rgb

---
 kwant/plotter.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/kwant/plotter.py b/kwant/plotter.py
index 036a0c6..8acef47 100644
--- a/kwant/plotter.py
+++ b/kwant/plotter.py
@@ -286,14 +286,14 @@ if mpl_enabled:
                     if (self._facecolors_orig is not None and
                         self._facecolors_orig.shape[0] > 1):
                         shape = list(self._facecolors_orig.shape)
-                        shape[0] = vs.shape[0]
+                        shape[0] = vs.shape[1]
                         super(Path3DCollection, self).set_facecolors(
                             np.resize(self._facecolors_orig, shape)[indx])
 
                     if (self._edgecolors_orig is not None and
                         self._edgecolors_orig.shape[0] > 1):
                         shape = list(self._edgecolors_orig.shape)
-                        shape[0] = vs.shape[0]
+                        shape[0] = vs.shape[1]
                         super(Path3DCollection, self).set_edgecolors(
                                                 np.resize(self._edgecolors_orig,
                                                           shape)[indx])
-- 
GitLab