[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

fuse/common.py

@@ -77,15 +77,13 @@ def full_array_to_numpy(array, dtype):
 def _sample_from_distribution(p, channel, spe_scaling_factor_distributions):
     """Function to sample from a SPE scaling factor distribution for a given
     channel."""
-
     indices = np.int64(p * 2000) + 1
     return spe_scaling_factor_distributions[channel, indices]
 
 
 @numba.njit()
 def sample_spe_scaling_factors(p, channel, spe_scaling_factor_distributions):
     """Function to sample the spe scaling factors for multiple photons."""
-
     result = []
     for i in range(len(p)):
         result.append(
@@ -269,7 +267,6 @@ def photon_gain_calculation(
 ):
     """Function to calculate the PMT gain a photon will be amplified with in
     the waveform simulation."""
-
     # Sample if the photon is a double PE emission
     _photon_is_dpe = rng.binomial(n=1, p=p_double_pe_emision, size=len(_photon_channels)).astype(
         np.bool_
@@ -316,7 +313,6 @@ def build_photon_propagation_output(
 
 def pmt_gains(to_pe, digitizer_voltage_range, digitizer_bits, pmt_circuit_load_resistor):
     """Build PMT Gains from PMT gain model and digitizer parameters."""
-
     adc_2_current = digitizer_voltage_range / 2 ** (digitizer_bits) / pmt_circuit_load_resistor
 
     gains = np.divide(

fuse/context.py

@@ -119,7 +119,6 @@ def microphysics_context(
     extra_plugins=[],
 ):
     """Function to create a fuse microphysics simulation context."""
-
     st = strax.Context(storage=output_folder, **common_opts)
     st.set_config(dict(check_raw_record_overlaps=True, **common_config))
 
@@ -159,7 +158,6 @@ def xenonnt_fuse_full_chain_simulation(
     This context includes all the necessary configs and plugins for the
     simulation.
     """
-
     # Load config file
     if run_without_config_file:
         # Just a dummy name to avoid errors. We use this to setup context for the docs.
@@ -308,7 +306,6 @@ def public_config_context(
     clustering_method="dbscan",
 ):
     """Create a context for the use of fuse with public XENONnT configs."""
-
     st = strax.Context(storage=output_folder, **straxen.contexts.common_opts)
     st.simulation_config_file = simulation_config_file
     st.config.update(dict(check_raw_record_overlaps=True, **straxen.contexts.common_config))

fuse/context_utils.py

@@ -73,7 +73,6 @@ def set_simulation_config_file(context, config_file_name):
 @URLConfig.register("pattern_map")
 def pattern_map(map_data, pmt_mask, method="WeightedNearestNeighbors"):
     """Pattern map handling."""
-
     if "compressed" in map_data:
         compressor, dtype, shape = map_data["compressed"]
         map_data["map"] = np.frombuffer(

fuse/plugins/detector_physics/delayed_electrons/photo_ionization_electrons.py

@@ -255,7 +255,6 @@ def ramdom_xy_position(n, radius, rng):
 # We should try to make it a general function in the future.
 def group_electrons_by_cluster_id(electrons):
     """Function to group electrons by cluster_id."""
-
     sort_index = stable_argsort(electrons["cluster_id"])
 
     electrons_sorted = electrons[sort_index]

fuse/plugins/detector_physics/secondary_scintillation.py

@@ -208,7 +208,6 @@ def get_s2_light_yield(self, positions):
             positions: 2d array of positions (floats) returns array
                 of floats (mean expectation)
         """
-
         if self.se_gain_from_map:
             sc_gain = self.se_gain_map(positions)
         else:
@@ -233,7 +232,6 @@ def get_s2_light_yield(self, positions):
 
 def group_result_photons_by_cluster_id(result, cluster_id):
     """Function to group result_photons by cluster_id."""
-
     sort_index = stable_argsort(cluster_id)
 
     cluster_id_sorted = cluster_id[sort_index]

fuse/plugins/micro_physics/cuts_and_selections/apply_selections.py

@@ -18,11 +18,11 @@
 class SelectionMerger(FuseBasePlugin):
     """Merge cuts/selections and stamp per-volume constants.
 
-    The selection logic is given as a string expression over boolean fields
-    in the `clustered_interactions` data. The expression may use '&', '|', '~',
-    and parentheses. For example, to select interactions in the fiducial volume
-    and with energy between 1 and 10 keV, use:
-        "volume_selection & energy_range_cut"
+    The selection logic is given as a string expression over boolean
+    fields in the `clustered_interactions` data. The expression may use
+    '&', '|', '~', and parentheses. For example, to select interactions
+    in the fiducial volume and with energy between 1 and 10 keV, use:
+    "volume_selection & energy_range_cut"
     """
 
     __version__ = "1.0.0"

fuse/plugins/micro_physics/find_cluster.py

@@ -81,7 +81,6 @@ def find_cluster(interactions, cluster_size_space, cluster_size_time):
 
         First interactions are clustered in time, then in space.
         """
-
         time_cluster = simple_1d_clustering(interactions["time"], cluster_size_time)
 
         # Splitting into time cluster and apply space clustering space:
@@ -127,7 +126,6 @@ def simple_1d_clustering(data, scale):
     Returns:
         clusters_undo_sort (np.array): Cluster Labels
     """
-
     idx_sort = stable_argsort(data)
     idx_undo_sort = stable_argsort(idx_sort)
 

fuse/plugins/micro_physics/input.py

@@ -256,7 +256,6 @@ def __init__(
 
     def output_chunk(self):
         """Function to return one chunk of data from the root or csv file."""
-
         if self.file_type == "root":
             interactions, n_simulated_events, start, stop = self._load_root_file()
         elif self.file_type == "csv":
@@ -586,7 +585,6 @@ def _load_csv_file(self):
             start: Index of the first loaded interaction
             stop: Index of the last loaded interaction
         """
-
         self.log.debug("Load instructions from a csv file!")
 
         df = pd.read_csv(self.file)
@@ -632,7 +630,6 @@ def _awkwardify_df(df):
         Returns:
             ak.Array(dictionary): awkward array
         """
-
         _, evt_offsets = np.unique(df["eventid"], return_counts=True)
 
         dictionary = {

fuse/plugins/micro_physics/lineage_cluster.py

@@ -90,7 +90,6 @@ def compute(self, geant4_interactions):
         Returns:
             np.ndarray: An array of cluster IDs with corresponding time and endtime values.
         """
-
         self.log.debug(f"Building lineages for {len(geant4_interactions)} interactions")
 
         if len(geant4_interactions) == 0:
@@ -288,10 +287,9 @@ def build_lineage_for_event(
                         )
 
                 else:
-                    raise ValueError(
-                        "There is no last particle interaction but we have seen \
-                        this particle before.... Makes no sense.."
-                    )
+                    raise ValueError( \
+                                     "There is no last particle interaction but we have seen \
+                        this particle before.... Makes no sense..")
 
         tmp_result["main_cluster_type"] = main_cluster_type
 
@@ -319,7 +317,6 @@ def get_particle(event_interactions, event_lineage, index, trackid_lookup):
 def get_last_particle_interaction(event_interactions, particle, particle_lineage):
     """Returns the last (previous in time) interaction of the particle that is
     in the lineage."""
-
     # Get all interactions for the given particle
     all_particle_interactions = event_interactions[
         event_interactions["trackid"] == particle["trackid"]
@@ -369,7 +366,6 @@ def get_parent(event_interactions, event_lineage, particle, parent_lookup):
 
 def is_particle_in_lineage(lineage):
     """Function to check if a particle is already in a lineage."""
-
     # All particles in the lineage have not been added to a lineage yet
     if np.all(lineage["lineage_index"] == 0):
         return False
@@ -384,7 +380,6 @@ def num_there(s):
 def classify_lineage(particle_interaction, classify_ic_as_gamma, classify_phot_as_beta):
     """Function to classify a new lineage based on the particle and its parent
     information."""
-
     def classify_gamma(particle_interaction):
         if particle_interaction["edproc"] == "compt":
             return NEST_BETA
@@ -475,7 +470,6 @@ def is_lineage_broken(
     time_threshold,
 ):
     """Function to check if the lineage is broken."""
-
     # second step of a decay. We want to split the lineage
     if (
         particle["creaproc"] == "RadioactiveDecayBase"
@@ -535,7 +529,6 @@ def is_lineage_broken(
 def get_element_and_mass(particle_type):
     """Function to get the element and the mass number from the particle
     type."""
-
     pattern_match = re.match(r"([a-z]+)([0-9]+)", particle_type, re.I)
 
     if pattern_match:

fuse/plugins/micro_physics/yields.py

@@ -129,10 +129,8 @@ def update_nest_width_parameters(self):
         if self.nest_width_parameters is not None:
             for key, value in self.nest_width_parameters.items():
                 if key not in parameters_key_map:
-                    raise ValueError(
-                        f"Unknown NEST width parameter {key}.\
-                        Available parameters: {parameters_key_map.keys()}"
-                    )
+                    raise ValueError(f"Unknown NEST width parameter {key}.\
+                        Available parameters: {parameters_key_map.keys()}")
                 self.log.debug(f"Updating NEST width parameter {key} to {value}")
                 free_parameters[parameters_key_map[key]] = value
 
@@ -183,15 +181,13 @@ def compute(self, interactions_in_roi):
 
     def get_quanta(self, en, model, e_field, A, Z, create_s2, density):
         """Function to get quanta for given parameters using NEST."""
-
         yields_result = self.get_yields_from_NEST(en, model, e_field, A, Z, density)
 
         return self.process_yields(yields_result, create_s2)
 
     def get_yields_from_NEST(self, en, model, e_field, A, Z, density):
         """Function which uses NEST to yield photons and electrons for a given
         set of parameters."""
-
         # Fix for Kr83m events
         max_allowed_energy_difference = 1  # keV
         if model == 11:
@@ -218,10 +214,8 @@ def get_yields_from_NEST(self, en, model, e_field, A, Z, density):
             e_field = self.fix_gamma_yield_field
 
         if e_field < 0:
-            raise ValueError(
-                f"Negative electric field {e_field} V/cm not allowed. \
-                (no error will be raised by NEST)."
-            )
+            raise ValueError(f"Negative electric field {e_field} V/cm not allowed. \
+                (no error will be raised by NEST).")
 
         yields_result = self.nc.GetYields(
             interaction=nestpy.INTERACTION_TYPE(model),
@@ -238,7 +232,6 @@ def get_yields_from_NEST(self, en, model, e_field, A, Z, density):
 
     def process_yields(self, yields_result, create_s2):
         """Process the yields with NEST to get actual quanta."""
-
         # Density argument is not used in function...
         event_quanta = self.nc.GetQuanta(
             yields_result, free_parameters=self.updated_nest_width_parameters