Skip to content

Commit

Permalink
Remove docstring overwrite; update DispReconstructor docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
LukasBeiske committed Nov 17, 2023
1 parent de277ba commit 027a9c8
Showing 1 changed file with 2 additions and 9 deletions.
11 changes: 2 additions & 9 deletions ctapipe/reco/sklearn.py
Original file line number Diff line number Diff line change
Expand Up @@ -367,9 +367,6 @@ class EnergyRegressor(SKLearnRegressionReconstructor):
property = ReconstructionProperty.ENERGY

def __call__(self, event: ArrayEventContainer) -> None:
"""
Apply model for a single event and fill result into the event container.
"""
for tel_id in event.trigger.tels_with_trigger:
table = collect_features(event, tel_id, self.instrument_table)
table = self.feature_generator(table, subarray=self.subarray)
Expand Down Expand Up @@ -398,7 +395,6 @@ def __call__(self, event: ArrayEventContainer) -> None:
self.stereo_combiner(event)

def predict_table(self, key, table: Table) -> Dict[ReconstructionProperty, Table]:
"""Predict on a table of events."""
table = self.feature_generator(table, subarray=self.subarray)

n_rows = len(table)
Expand Down Expand Up @@ -436,9 +432,6 @@ class ParticleClassifier(SKLearnClassificationReconstructor):
property = ReconstructionProperty.PARTICLE_TYPE

def __call__(self, event: ArrayEventContainer) -> None:
"""
Apply model for a single event and fill result into the event container.
"""
for tel_id in event.trigger.tels_with_trigger:
table = collect_features(event, tel_id, self.instrument_table)
table = self.feature_generator(table, subarray=self.subarray)
Expand All @@ -465,7 +458,6 @@ def __call__(self, event: ArrayEventContainer) -> None:
self.stereo_combiner(event)

def predict_table(self, key, table: Table) -> Dict[ReconstructionProperty, Table]:
"""Predict on a table of events."""
table = self.feature_generator(table, subarray=self.subarray)

n_rows = len(table)
Expand All @@ -492,7 +484,8 @@ def predict_table(self, key, table: Table) -> Dict[ReconstructionProperty, Table

class DispReconstructor(Reconstructor):
"""
Predict absolute value and sign for disp origin reconstruction for each telescope.
Predict absolute value and sign for disp origin reconstruction and
convert to altitude and azimuth prediction for each telescope.
"""

target = "true_disp"
Expand Down

0 comments on commit 027a9c8

Please sign in to comment.