Update ctapipe traitlets calls in Franca's tools

cta-observatory · Aug 22, 2024 · bca2f95 · bca2f95
1 parent 8d019d8
commit bca2f95
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/nectarchain/tools/write_camera_calibration.py b/src/nectarchain/tools/write_camera_calibration.py
@@ -42,11 +42,9 @@ class CalibrationHDF5Writer(Tool):
         800, help="Temporary cut on charge to eliminate events without led signal"
     ).tag(config=True)
 
-    pedestal_product = traits.enum_trait(
-        PedestalCalculator, default="PedestalIntegrator"
-    )
+    pedestal_product = traits.Enum(PedestalCalculator, default="PedestalIntegrator")
 
-    flatfield_product = traits.enum_trait(
+    flatfield_product = traits.Enum(
         FlatFieldCalculator, default="FlasherFlatFieldCalculator"
     )
 

diff --git a/src/nectarchain/tools/write_pedestals.py b/src/nectarchain/tools/write_pedestals.py
@@ -24,9 +24,7 @@ class PedestalHDF5Writer(Tool):
         config=True
     )
 
-    calculator_product = traits.enum_trait(
-        PedestalCalculator, default="PedestalIntegrator"
-    )
+    calculator_product = traits.Enum(PedestalCalculator, default="PedestalIntegrator")
 
     aliases = Dict(
         dict(