standardize astropy table import and remove dead comments

lsst-dm · Sep 24, 2024 · b3fe18e · b3fe18e
1 parent 65bab23
commit b3fe18e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 10 deletions.
diff --git a/python/lsst/meas/pz/estimate_pz_task.py b/python/lsst/meas/pz/estimate_pz_task.py
@@ -34,7 +34,6 @@
 from ceci.config import StageConfig as CeciStageConfig
 from ceci.config import StageParameter as CeciParam
 
-# from ceci.stage import PipelineStage as CeciPipelineStage
 from lsst.pipe.base import (
     PipelineTask,
     PipelineTaskConfig,
@@ -43,7 +42,7 @@
     Task,
 )
 
-import astropy.table as atable
+from astropy.table import Table
 from rail.core.model import Model
 from rail.estimation.estimator import CatEstimator
 from rail.interfaces import PZFactory
@@ -64,7 +63,6 @@ class EstimatePZConnections(
         storageClass="PZModel",
         dimensions=["instrument"],
         isCalibration=True,
-        # lookupFunction=_pzModelLookup,
     )
 
     objectTable = cT.Input(
@@ -324,14 +322,14 @@ def _get_mag_err_names(self) -> dict[str, str]:
 
     def _get_mags_and_errs(
         self,
-        fluxes: atable,
+        fluxes: Table,
         mag_offset: float,
     ) -> dict[str, np.array]:
         """Fill and return a numpy dict with mags and mag errors
 
         Parameters
         ----------
-        fluxes : atable
+        fluxes : Table
             Input fluxes and flux errors
 
         mag_offset : float
@@ -411,13 +409,13 @@ def col_names(
 
     def run(
         self,
-        fluxes: atable,
+        fluxes: Table,
     ) -> Struct:
         """Run a p(z) estimation algorithm
 
         Parameters
         ----------
-        fluxes: atable
+        fluxes: Table
             Fluxes used to compute the redshifts
 
         Returns
@@ -485,7 +483,7 @@ def runQuantum(self, butlerQC, inputRefs, outputRefs):
     def run(
         self,
         pzModel: Model,
-        fluxes: atable,
+        fluxes: Table,
         skip_init: bool = False,
     ) -> Struct:
         if not skip_init:

diff --git a/python/lsst/meas/pz/estimate_pz_task_trainz.py b/python/lsst/meas/pz/estimate_pz_task_trainz.py
@@ -27,7 +27,7 @@
 ]
 
 import numpy as np
-import astropy.table as atable
+from astropy.table import Table
 from rail.estimation.estimator import CatEstimator
 from rail.estimation.algos.train_z import TrainZEstimator
 
@@ -72,7 +72,7 @@ class EstimatePZTrainZAlgoTask(EstimatePZAlgoTask):
 
     def _get_mags_and_errs(
         self,
-        fluxes: atable,
+        fluxes: Table,
         mag_offset: float,
     ) -> dict[str, np.array]: