Make sure model build doesn't fail if rings == 1

This commit is contained in:
Paul Romano 2019-12-19 14:59:44 -06:00
parent 6448eaff09
commit b5b6f4611a
2 changed files with 8 additions and 2 deletions

View file

@ -44,7 +44,10 @@ else:
directory.mkdir(exist_ok=True)
# Define geometry with a single assembly
ring_radii = np.sqrt(np.arange(1, args.rings)*pellet_OR**2 / args.rings)
if args.rings > 1:
ring_radii = np.sqrt(np.arange(1, args.rings)*pellet_OR**2 / args.rings)
else:
ring_radii = None
assembly = assembly_universes(ring_radii, args.axial, args.depleted)
lattice_sides = openmc.model.get_rectangular_prism(lattice_pitch, lattice_pitch,
boundary_type='reflective')

View file

@ -51,7 +51,10 @@ else:
directory = args.output_dir
directory.mkdir(exist_ok=True)
ring_radii = np.sqrt(np.arange(1, args.rings)*pellet_OR**2 / args.rings)
if args.rings > 1:
ring_radii = np.sqrt(np.arange(1, args.rings)*pellet_OR**2 / args.rings)
else:
ring_radii = None
geometry = core_geometry(ring_radii, args.axial, args.depleted)
#### "Differentiate" the geometry if using distribmats