From 48081d8b35b24eb690bee05dd3371737821c8e9b Mon Sep 17 00:00:00 2001 From: Eric Charles Date: Wed, 26 Jun 2024 13:27:42 -0700 Subject: [PATCH] Put back a couple of includes that ruff removed b/c it though they were unused --- src/rail/core/utils.py | 4 +++- src/rail/estimation/estimator.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/src/rail/core/utils.py b/src/rail/core/utils.py index 8e506efc..fef59270 100644 --- a/src/rail/core/utils.py +++ b/src/rail/core/utils.py @@ -2,7 +2,9 @@ # to move RAILDIR and find_rail_file to utils.path_utils from rail.utils.path_utils import find_rail_file +from rail.utils.path_utils import RAILDIR __all__ = [ - 'find_rail_file' + 'find_rail_file', + 'RAILDIR', ] diff --git a/src/rail/estimation/estimator.py b/src/rail/estimation/estimator.py index 0febaa40..fee3e9a9 100644 --- a/src/rail/estimation/estimator.py +++ b/src/rail/estimation/estimator.py @@ -11,7 +11,7 @@ from rail.core.point_estimation import PointEstimationMixin # for backwards compatibility, to avoid break stuff that imports it from here - +from rail.estimation.informer import CatInformer # pylint: disable=unused-import class CatEstimator(RailStage, PointEstimationMixin): """The base class for making photo-z posterior estimates from catalog-like inputs