Try to fix tests

cta-observatory · Sep 27, 2023 · ea94f71 · ea94f71
1 parent 89cac19
commit ea94f71
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/magicctapipe/scripts/lst1_magic/lst1_magic_dl1_stereo_to_dl2.py b/magicctapipe/scripts/lst1_magic/lst1_magic_dl1_stereo_to_dl2.py
@@ -38,7 +38,12 @@
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.StreamHandler())
 logger.setLevel(logging.INFO)
-
+TEL_COMBINATIONS = {
+    "M1_M2": [2, 3],  # combo_type = 0
+    "LST1_M1": [1, 2],  # combo_type = 1
+    "LST1_M2": [1, 3],  # combo_type = 2
+    "LST1_M1_M2": [1, 2, 3],  # combo_type = 3
+}  #####TO BE REMOVED WHEN SWITCHING TO THE NEW RFs IMPLEMENTTATION (1 RF PER TELESCOPE) !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
 def apply_rfs(event_data, estimator):
     """

diff --git a/magicctapipe/scripts/lst1_magic/lst1_magic_train_rfs.py b/magicctapipe/scripts/lst1_magic/lst1_magic_train_rfs.py
@@ -41,7 +41,7 @@
 import pandas as pd
 import yaml
 from magicctapipe.io import format_object, load_train_data_files
-from magicctapipe.io.io import GROUP_INDEX_TRAIN, TEL_NAMES
+from magicctapipe.io.io import GROUP_INDEX_TRAIN
 from magicctapipe.reco import DispRegressor, EnergyRegressor, EventClassifier
 
 __all__ = [
@@ -54,6 +54,8 @@
 logger = logging.getLogger(__name__)
 logger.addHandler(logging.StreamHandler())
 logger.setLevel(logging.INFO)
+TEL_NAMES = {1: "LST-1", 2: "MAGIC-I", 3: "MAGIC-II"} #####TO BE REMOVED WHEN SWITCHING TO THE NEW RFs IMPLEMENTTATION (1 RF PER TELESCOPE) !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
+
 
 # True event class of gamma and proton MCs
 EVENT_CLASS_GAMMA = 0