forked from crp/ecp-benchmarks
Make sure model build doesn't fail if rings == 1
This commit is contained in:
parent
6448eaff09
commit
b5b6f4611a
2 changed files with 8 additions and 2 deletions
|
|
@ -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')
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue