Merge pull request #31 from fusion-energy/working_with_SourceBase

extending sourcebase instead of source class
fusion-energy · Aug 16, 2023 · 82656ef · 82656ef
2 parents 58e8159 + eeed164
commit 82656ef
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/openmc_source_plotter/core.py b/src/openmc_source_plotter/core.py
@@ -217,7 +217,7 @@ def plot_source_direction(
 also provided for convenience. Additional methods are plot_source_energy(),
 plot_source_position(), plot_source_direction(), sample_initial_particles()
 """
-openmc.Source.sample_initial_particles = sample_initial_particles
-openmc.Source.plot_source_energy = plot_source_energy
-openmc.Source.plot_source_position = plot_source_position
-openmc.Source.plot_source_direction = plot_source_direction
+openmc.SourceBase.sample_initial_particles = sample_initial_particles
+openmc.SourceBase.plot_source_energy = plot_source_energy
+openmc.SourceBase.plot_source_position = plot_source_position
+openmc.SourceBase.plot_source_direction = plot_source_direction