jan/gridlock
2
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fix use of enum for surface_normal for visualize_slice

Browse Source
This commit is contained in:
jan 2016-07-13 16:09:57 -07:00
parent 1e4ee5e057
commit c8cbf26a8c
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
gridlock/grid.py
View File

@ -776,6 +776,10 @@ class Grid(object):
""" """
from matplotlib import pyplot 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, grid_slice = self.get_slice(surface_normal=surface_normal,
center=center, center=center,
which_shifts=which_shifts, which_shifts=which_shifts,