diff --git a/disruption_py/shots/parameter_methods/cmod/basic_parameter_methods.py b/disruption_py/shots/parameter_methods/cmod/basic_parameter_methods.py index e4812524..33c41855 100644 --- a/disruption_py/shots/parameter_methods/cmod/basic_parameter_methods.py +++ b/disruption_py/shots/parameter_methods/cmod/basic_parameter_methods.py @@ -1159,18 +1159,17 @@ def _get_peaking_factors(params: ShotDataRequestParams): @staticmethod @parameter_cached_method( columns=["prad_peaking"], - used_trees=["cmod", "spectroscopy"], + used_trees=["_efit_tree", "spectroscopy"], tokamak=Tokamak.CMOD, ) def _get_prad_peaking(params: ShotDataRequestParams): prad_peaking = np.full(len(params.shot_props.times), np.nan) try: - # TODO: why use CMOD here? r0 = 0.01 * params.mds_conn.get_data( - r"\efit_aeqdsk:rmagx", tree_name="cmod" + r"\efit_aeqdsk:rmagx", tree_name="_efit_tree" ) z0 = 0.01 * params.mds_conn.get_data( - r"\efit_aeqdsk:zmagx", tree_name="cmod" + r"\efit_aeqdsk:zmagx", tree_name="_efit_tree" ) aminor, efit_time = params.mds_conn.get_data_with_dims( r"\efit_aeqdsk:aminor", tree_name="_efit_tree"