From b271ce98698d509c3b5309415f72c8a924485338 Mon Sep 17 00:00:00 2001 From: Bastien Lacave Date: Thu, 23 Oct 2025 14:50:38 +0200 Subject: [PATCH 1/2] remove newaxis --- ctlearn/tools/predict_LST1.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ctlearn/tools/predict_LST1.py b/ctlearn/tools/predict_LST1.py index 4a385e57..cf695918 100644 --- a/ctlearn/tools/predict_LST1.py +++ b/ctlearn/tools/predict_LST1.py @@ -604,7 +604,7 @@ def start(self): colname, colname.replace("_tel", "") ) classification_subarray_table.add_column( - classification_is_valid[np.newaxis], name=f"{self.prefix}_telescopes" + classification_is_valid, name=f"{self.prefix}_telescopes" ) # Save the prediction to the output file write_table( @@ -685,7 +685,7 @@ def start(self): colname, colname.replace("_tel", "") ) energy_subarray_table.add_column( - energy_is_valid[np.newaxis], name=f"{self.prefix}_telescopes" + energy_is_valid, name=f"{self.prefix}_telescopes" ) # Save the prediction to the output file write_table( @@ -803,7 +803,7 @@ def start(self): colname, colname.replace("_tel", "") ) direction_subarray_table.add_column( - direction_is_valid[np.newaxis], name=f"{self.prefix}_telescopes" + direction_is_valid, name=f"{self.prefix}_telescopes" ) # Save the prediction to the output file write_table( From 0153838e23386863aeb3bd3103827ecc0ea83723 Mon Sep 17 00:00:00 2001 From: Bastien Lacave Date: Thu, 23 Oct 2025 15:11:23 +0200 Subject: [PATCH 2/2] put values in arrays --- ctlearn/tools/predict_LST1.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ctlearn/tools/predict_LST1.py b/ctlearn/tools/predict_LST1.py index cf695918..e27e68c6 100644 --- a/ctlearn/tools/predict_LST1.py +++ b/ctlearn/tools/predict_LST1.py @@ -604,7 +604,7 @@ def start(self): colname, colname.replace("_tel", "") ) classification_subarray_table.add_column( - classification_is_valid, name=f"{self.prefix}_telescopes" + [[val] for val in classification_is_valid], name=f"{self.prefix}_telescopes" ) # Save the prediction to the output file write_table( @@ -685,7 +685,7 @@ def start(self): colname, colname.replace("_tel", "") ) energy_subarray_table.add_column( - energy_is_valid, name=f"{self.prefix}_telescopes" + [[val] for val in energy_is_valid], name=f"{self.prefix}_telescopes" ) # Save the prediction to the output file write_table( @@ -803,7 +803,7 @@ def start(self): colname, colname.replace("_tel", "") ) direction_subarray_table.add_column( - direction_is_valid, name=f"{self.prefix}_telescopes" + [[val] for val in direction_is_valid], name=f"{self.prefix}_telescopes" ) # Save the prediction to the output file write_table(