diff --git a/src/rail/interfaces/pz_factory.py b/src/rail/interfaces/pz_factory.py index 0e0da4d..196bd70 100644 --- a/src/rail/interfaces/pz_factory.py +++ b/src/rail/interfaces/pz_factory.py @@ -97,7 +97,7 @@ def run_cat_estimator_stage( return stage_obj.estimate(handle) @staticmethod - def evaluate_single_pz( + def estimate_single_pz( stage_obj: CatEstimator, data_table: dict, input_size: int=1, diff --git a/tests/interfaces/test_pz_factory.py b/tests/interfaces/test_pz_factory.py index 0339a14..77f24f4 100644 --- a/tests/interfaces/test_pz_factory.py +++ b/tests/interfaces/test_pz_factory.py @@ -18,7 +18,7 @@ def test_pz_factory(): input_file = find_rail_file('examples_data/testdata/validation_10gal.hdf5') - out_single = PZFactory.evaluate_single_pz(stage, {'d':np.array([1,1])}) + out_single = PZFactory.estimate_single_pz(stage, {'d':np.array([1,1])}) assert out_single.npdf == 1 out_handle = PZFactory.run_cat_estimator_stage(