From c8cbf26a8c864bc5f5c5d55ddd88c5bfd606f0bf Mon Sep 17 00:00:00 2001 From: jan Date: Wed, 13 Jul 2016 16:09:57 -0700 Subject: [PATCH] Fix use of enum for surface_normal for visualize_slice --- gridlock/grid.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/gridlock/grid.py b/gridlock/grid.py index f64c740..5849339 100644 --- a/gridlock/grid.py +++ b/gridlock/grid.py @@ -776,6 +776,10 @@ class Grid(object): """ from matplotlib import pyplot + # Set surface normal to its integer value + if isinstance(surface_normal, Direction): + surface_normal = surface_normal.value + grid_slice = self.get_slice(surface_normal=surface_normal, center=center, which_shifts=which_shifts,