From d084b718f00b9d4cb6d22e2842cf31bc51ff9edb Mon Sep 17 00:00:00 2001 From: Paul Romano Date: Sat, 4 Mar 2017 00:25:30 -0600 Subject: [PATCH] Fix Geometry.get_all_lattices() and Geometry.get_all_material_universes() --- openmc/geometry.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/openmc/geometry.py b/openmc/geometry.py index 53a7ffa28e..079647b79b 100644 --- a/openmc/geometry.py +++ b/openmc/geometry.py @@ -230,8 +230,8 @@ class Geometry(object): """ material_universes = OrderedDict() - for universe in self.get_all_universes(): - for cell in universe.cells: + for universe in self.get_all_universes().values(): + for cell in universe.cells.values(): if cell.fill_type in ('material', 'distribmat', 'void'): if universe not in material_universes: material_universes[universe.id] = universe @@ -249,7 +249,7 @@ class Geometry(object): """ lattices = OrderedDict() - for cell in self.get_all_cells(): + for cell in self.get_all_cells().values(): if cell.fill_type == 'lattice': if cell.fill not in lattices: lattices[cell.fill.id] = cell.fill