Add imports

antea/mcsim/sensor_functions.py

@@ -1,6 +1,8 @@
 import pandas as pd
 import numpy  as np
 
+from typing     import Sequence
+
 def apply_charge_fluctuation(sns_df: pd.DataFrame, DataSiPM_idx: pd.DataFrame):
     """
     Apply a fluctuation in the total detected charge, sensor by sensor,

antea/mcsim/sensor_functions_test.py

@@ -1,15 +1,18 @@
 import os
 import pandas as pd
+import numpy  as np
 
 from .. database import load_db as db
 
 from .. io.mc_io         import load_mcsns_response
+from .. io.mc_io         import load_mcTOFsns_response
 from .  sensor_functions import apply_charge_fluctuation
 from .  sensor_functions import apply_sipm_pde
 from .  sensor_functions import apply_sipm_saturation
 
 import hypothesis.strategies as st
 from hypothesis  import given
+from typing      import Sequence
 
 def test_number_of_sensors_is_the_same(ANTEADATADIR):
     """
@@ -62,7 +65,7 @@ def test_apply_sipm_saturation(ANTEADATADIR):
     is never higher than the original charge.
     '''
     PATH_IN       = os.path.join(ANTEADATADIR, 'petit_mc_fbk_test.pet.h5')
-    sns_response  = load_mcsns_response(PATH_IN)
+    sns_response  = load_mcTOFsns_response(PATH_IN)
     events        = sns_response.event_id.unique()
 
     evt = events[0]