Merge branch 'improved_energy_distribution' of github.com:fusion-ener…

…gy/openmc-plasma-source into improved_energy_distribution

src/openmc_plasma_source/fuel_types.py

@@ -49,7 +49,7 @@ def get_neutron_energy_distribution(
     E_pspec = np.linspace(0, 12, num_of_vals)  # E_pspec is exspected in MeV units
 
     DTmean, DTvar = nst.DTprimspecmoments(ion_temperature_kev)
-    print('DTmean', DTmean)
+    print("DTmean", DTmean)
     DDmean, DDvar = nst.DDprimspecmoments(ion_temperature_kev)
 
     if ["D", "T"] == sorted(set(fuel.keys())):
@@ -67,7 +67,11 @@ def get_neutron_energy_distribution(
         dd_source = openmc.stats.muir(e0=2.5e6, m_rat=4, kt=ion_temperature)
         dt_source = openmc.stats.muir(e0=14.06e6, m_rat=5, kt=ion_temperature)
         # todo look into combining distributions openmc.data.combine_distributions()
-        return [tt_source, dd_source, dt_source], [strength_TT, strength_DD, strength_DT]
+        return [tt_source, dd_source, dt_source], [
+            strength_TT,
+            strength_DD,
+            strength_DT,
+        ]
 
     elif ["D"] == sorted(set(fuel.keys())):
 
@@ -81,4 +85,3 @@ def get_neutron_energy_distribution(
         dNdE_TT = strength_TT * nst.dNdE_TT(E_pspec, ion_temperature_kev)
         tt_source = openmc.stats.Tabular(E_pspec * 1e6, dNdE_TT)
         return [tt_source], [strength_TT]
-