From 5d177a5ace4e640a76f5941b76468f3c604e938c Mon Sep 17 00:00:00 2001 From: Jonathan Shimwell Date: Wed, 26 Oct 2022 20:21:06 +0100 Subject: [PATCH] Apply suggestions from code review by @paulromano Co-authored-by: Paul Romano --- openmc/source.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/openmc/source.py b/openmc/source.py index 05954da777..7be7b838aa 100644 --- a/openmc/source.py +++ b/openmc/source.py @@ -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