Skip to content

Commit

Permalink
Merge pull request #19 from LSSTDESC/ceci2
Browse files Browse the repository at this point in the history
Update for ceci version 2
  • Loading branch information
hangqianjun authored Jul 17, 2024
2 parents 766f70e + ebb3199 commit a15ee8d
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ dynamic = ["version"]

dependencies = [
"scikit-learn",
"pz-rail-base",
"pz-rail-base>=1.0.3",
"qp-prob[full]",
]

Expand Down
8 changes: 4 additions & 4 deletions src/rail/estimation/algos/gpz.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,10 +73,10 @@ class GPzInformer(CatInformer):
replace_error_vals=Param(list, default_err_repl, msg="list of values to replace negative and nan mag err values")
)

def __init__(self, args, comm=None):
def __init__(self, args, **kwargs):
""" Constructor
Do CatInformer specific initialization"""
CatInformer.__init__(self, args, comm=comm)
super().__init__(args, **kwargs)
self.zgrid = None

def run(self):
Expand Down Expand Up @@ -144,10 +144,10 @@ class GPzEstimator(CatEstimator):
replace_error_vals=Param(list, default_err_repl, msg="list of values to replace negative and nan mag err values")
)

def __init__(self, args, comm=None):
def __init__(self, args, **kwargs):
""" Constructor:
Do CatEstimator specific initialization """
CatEstimator.__init__(self, args, comm=comm)
super().__init__(args, **kwargs)
self.zgrid = None
# check that lengths of bands, err_bands, and replace_error_vals match
if not np.logical_and(len(self.config.bands) == len(self.config.err_bands),
Expand Down

0 comments on commit a15ee8d

Please sign in to comment.