Fix use of enum for surface_normal for visualize_slice

2016-07-13 16:09:57 -07:00 · 2016-07-13 16:09:57 -07:00 · c8cbf26a8c
commit c8cbf26a8c
parent 1e4ee5e057
1 changed files with 4 additions and 0 deletions
--- 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,