Rewrite constructors for ceci v2

LSSTDESC · Jul 11, 2024 · 134007b · 134007b
1 parent 8ebd07c
commit 134007b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/rail/yaw_rail/correlation.py b/src/rail/yaw_rail/correlation.py
@@ -88,8 +88,8 @@ class YawBaseCorrelate(YawRailStage):
     inputs: list[tuple[str, YawCacheHandle]]
     outputs: list[tuple[str, YawCorrFuncHandle]]
 
-    def __init__(self, args, comm=None):
-        super().__init__(args, comm=comm)
+    def __init__(self, args, **kwargs):
+        super().__init__(args, **kwargs)
         self.yaw_config = Configuration.create(**self.get_algo_config_dict())
 
     @abstractmethod

diff --git a/src/rail/yaw_rail/summarize.py b/src/rail/yaw_rail/summarize.py
@@ -143,8 +143,8 @@ class YawSummarize(
         ("yaw_cc", YawRedshiftDataHandle),
     ]
 
-    def __init__(self, args, comm=None):
-        super().__init__(args, comm=comm)
+    def __init__(self, args, **kwargs):
+        super().__init__(args, **kwargs)
         config = {p: self.config_options[p].value for p in config_yaw_resampling}
         self.yaw_config = ResamplingConfig.create(**config)