File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 88
99import numpy as np
1010from astropy import units as u
11- from astropy .table import QTable
11+ from astropy .table import QTable , Table
1212from pyirf .binning import OVERFLOW_INDEX , UNDERFLOW_INDEX , calculate_bin_indices
1313from pyirf .spectral import (
1414 calculate_event_weights ,
1515)
1616
1717from ..core import Component , traits
18- from ..core .feature_generator import _shallow_copy_table
18+ from ..core .feature_generator import shallow_copy_table
1919from .binning import DefaultFoVOffsetBins
2020from .spectra import Spectra , spectrum_from_name
2121
@@ -69,10 +69,10 @@ def _compute_weights(self, events_table: QTable):
6969 f"{ self .__class__ .__name__ } weighting is not implemented"
7070 )
7171
72- def __call__ (self , events_table : QTable ) -> QTable :
72+ def __call__ (self , events_table : Table | QTable ) -> QTable :
7373 """Returns shallow copy of input table with a ``weight`` column added"""
7474
75- table = _shallow_copy_table (events_table )
75+ table = shallow_copy_table (events_table , output_cls = QTable )
7676 self ._compute_weights (table )
7777 return table
7878
You can’t perform that action at this time.
0 commit comments