fix up unit tests

LSSTDESC · Apr 8, 2024 · 8d154f8 · 8d154f8
1 parent 3afdeaf
commit 8d154f8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/evaluation/test_evaluation.py b/tests/evaluation/test_evaluation.py
@@ -9,7 +9,7 @@
 from rail.core.data import QPHandle, TableHandle,  QPOrTableHandle
 from rail.core.stage import RailStage
 from rail.core.utils import find_rail_file
-from rail.evaluation.evaluator import Evaluator
+from rail.evaluation.evaluator import OldEvaluator
 from rail.evaluation.dist_to_dist_evaluator import DistToDistEvaluator
 from rail.evaluation.dist_to_point_evaluator import DistToPointEvaluator
 from rail.evaluation.point_to_point_evaluator import PointToPointEvaluator
@@ -85,7 +85,7 @@ def test_evaluation_stage():
     pdf = DS.add_data("pdf", pdf_ens, QPHandle)
     truth_table = dict(redshift=zspec)
     truth = DS.add_data("truth", truth_table, TableHandle)
-    evaluator = Evaluator.make_stage(name="Eval")
+    evaluator = OldEvaluator.make_stage(name="Eval")
     evaluator.evaluate(pdf, truth)
 
     os.remove(