Browse Source

Fix use of enum for surface_normal for visualize_slice

tags/v0.3
jan 4 years ago
parent
commit
c8cbf26a8c
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      gridlock/grid.py

+ 4
- 0
gridlock/grid.py View File

@@ -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,


Loading…
Cancel
Save