Skip to content

Commit

Permalink
trying to fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Elisa-Visentin committed Sep 27, 2023
1 parent 37027e3 commit 89cac19
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions magicctapipe/reco/estimators.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,14 @@
import numpy as np
import pandas as pd
import sklearn.ensemble
from magicctapipe.io.io import TEL_NAMES


__all__ = ["EnergyRegressor", "DispRegressor", "EventClassifier"]

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) !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!


class EnergyRegressor:
Expand Down Expand Up @@ -60,7 +61,7 @@ def fit(self, event_data):
event_data: pandas.core.frame.DataFrame
Data frame of shower events
"""

self.telescope_rfs.clear()

# Loop over every telescope
Expand Down

0 comments on commit 89cac19

Please sign in to comment.