0
0
Просмотр исходного кода

Update compton_combiner.py

Fixed bug that the last run was included into the selection.
compton 1 год назад
Родитель
Сommit
7be5303036
1 измененных файлов с 4 добавлено и 4 удалено
  1. 4 4
      compton_combiner.py

+ 4 - 4
compton_combiner.py

@@ -20,7 +20,7 @@ import pandas as pd
 from tqdm import tqdm
 
 SEASONS = {
-    'name': ['RHO2013', 'BRK2013/16', 'HIGH2017', 'RHO2018', 'HIGH2019', 'LOW2020', 'HIGH2020', 'HIGH2021', 'NNBAR2021', 'HIGH2023'],
+    'name': ['RHO2013', 'BRK2013/16', 'HIGH2017', 'RHO2018', 'HIGH2019', 'LOW2020', 'HIGH2020', 'HIGH2021', 'NNBAR2022', 'HIGH2023'],
     'start_run': [18809, 32076, 36872, 48938, 70014, 85224, 89973, 98116, 107342, 131913, None],
 }
 
@@ -46,7 +46,7 @@ class RunsDBHandler():
         ----------
         range : Union[Tuple[int, Optional[int]], Tuple[datetime, datetime]]
             selection range
-            int range defines an interval in runs
+            int range defines an interval in runs (first included, last excluded)
             datetime range defines a time interval (NSK: +7:00 time)
         energy_point : Optional[float]
             energy point name, MeV (default is  None)
@@ -58,7 +58,7 @@ class RunsDBHandler():
         if isinstance(range[0], int):
             cond = f" AND run >= {range[0]} "
             if range[1] is not None:
-                cond += f" AND run <= {range[1]} "
+                cond += f" AND run < {range[1]} "
         elif isinstance(range[0], datetime):
             cond = f" AND starttime >= %s "
             if range[1] is not None:
@@ -466,7 +466,7 @@ def process_combined(combined_df: pd.DataFrame, runs_df: pd.DataFrame, compton_d
         spread_mean=('spread_mean', 'min'), spread_mean_test=('spread_mean', 'max'),
         spread_std=('spread_std', 'min'), spread_std_test=('spread_std', 'max'),
     ).reset_index().set_index('point_idx')
-    #return combined_df
+    # return combined_df
     
     result_df = pd.DataFrame(columns=['energy_point', 'first_run', 'last_run', 'mean_energy', 'mean_energy_stat_err', 'mean_energy_sys_err', 'mean_spread', 'mean_spread_stat_err', 'used_lum', 'comment'])