Apply suggestions from code review by @paulromano

Co-authored-by: Paul Romano <paul.k.romano@gmail.com>
This commit is contained in:
Jonathan Shimwell 2022-10-26 20:21:06 +01:00 committed by GitHub
parent 4116f6cda0
commit 5d177a5ace
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -364,8 +364,8 @@ class SourceParticle:
"""
def __init__(
self,
r: Tuple[float, float, float] = (0., 0., 0.),
u: Tuple[float, float, float] = (0., 0., 1.),
r: typing.Iterable[float] = (0., 0., 0.),
u: typing.Iterable[float] = (0., 0., 1.),
E: float = 1.0e6,
time: float = 0.0,
wgt: float = 1.0,
@ -401,7 +401,7 @@ class SourceParticle:
def write_source_file(
source_particles: 'typing.Iterable[openmc.SourceParticle]',
source_particles: typing.Iterable[SourceParticle],
filename: PathLike, **kwargs
):
"""Write a source file using a collection of source particles