manage to keep the type hinting

LIBRA-project · Dec 19, 2024 · 697dfd4 · 697dfd4
1 parent 72e00dd
commit 697dfd4
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/libra_toolbox/neutronics/neutron_source.py b/libra_toolbox/neutronics/neutron_source.py
@@ -6,8 +6,17 @@
 import pandas as pd
 import numpy as np
 
+try:
+    import h5py
+    import openmc
+    from openmc import IndependentSource
+except ModuleNotFoundError:
+    pass
 
-def A325_generator_diamond(center=(0, 0, 0), reference_uvw=(0, 0, 1)) -> Iterable:
+
+def A325_generator_diamond(
+    center=(0, 0, 0), reference_uvw=(0, 0, 1)
+) -> "Iterable[IndependentSource]":
     """
     Builds the MIT-VaultLab A-325 neutron generator in OpenMC
     with data tabulated from John Ball and Shon Mackie characterization