From e54ec7c88fa8ad7d63e05b45a194be71d76044d7 Mon Sep 17 00:00:00 2001 From: Lukas Beiske Date: Fri, 17 Nov 2023 13:04:08 +0100 Subject: [PATCH] Remove docstring overwrite; update DispReconstructor docstring --- ctapipe/reco/sklearn.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/ctapipe/reco/sklearn.py b/ctapipe/reco/sklearn.py index a128f8db77a..bb8e9b8af0c 100644 --- a/ctapipe/reco/sklearn.py +++ b/ctapipe/reco/sklearn.py @@ -379,9 +379,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) @@ -410,7 +407,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) @@ -448,9 +444,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) @@ -477,7 +470,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) @@ -504,7 +496,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"