Skip to content

Commit 42ce382

Browse files
Merge pull request #234 from neutrinoceros/rfc_analysis_cb
2 parents a99296e + c0d5b2a commit 42ce382

2 files changed

Lines changed: 3 additions & 21 deletions

File tree

yt_astro_analysis/halo_analysis/halo_catalog/analysis_operators.py

Lines changed: 2 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -21,14 +21,10 @@ class AnalysisCallback:
2121
to the target object, write out data, etc, but does not return anything.
2222
"""
2323

24-
def __init__(self, function, args=None, kwargs=None):
24+
def __init__(self, function, *args, **kwargs):
2525
self.function = function
2626
self.args = args
27-
if self.args is None:
28-
self.args = []
2927
self.kwargs = kwargs
30-
if self.kwargs is None:
31-
self.kwargs = {}
3228

3329
def __call__(self, target):
3430
self.function(target, *self.args, **self.kwargs)
@@ -49,9 +45,6 @@ class AnalysisFilter(AnalysisCallback):
4945
whether analysis is continued.
5046
"""
5147

52-
def __init__(self, function, *args, **kwargs):
53-
AnalysisCallback.__init__(self, function, args, kwargs)
54-
5548
def __call__(self, target):
5649
return self.function(target, *self.args, **self.kwargs)
5750

@@ -69,9 +62,6 @@ class AnalysisQuantity(AnalysisCallback):
6962
performs some analysis, and then returns a value.
7063
"""
7164

72-
def __init__(self, function, *args, **kwargs):
73-
AnalysisCallback.__init__(self, function, args, kwargs)
74-
7565
def __call__(self, target):
7666
return self.function(target, *self.args, **self.kwargs)
7767

@@ -89,14 +79,10 @@ class AnalysisRecipe:
8979
adds a series of callbacks, quantities, and filters.
9080
"""
9181

92-
def __init__(self, function, args=None, kwargs=None):
82+
def __init__(self, function, *args, **kwargs):
9383
self.function = function
9484
self.args = args
95-
if self.args is None:
96-
self.args = []
9785
self.kwargs = kwargs
98-
if self.kwargs is None:
99-
self.kwargs = {}
10086

10187
def __call__(self, pipeline):
10288
return self.function(pipeline, *self.args, **self.kwargs)

yt_astro_analysis/halo_analysis/halo_catalog/halo_finding_methods.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,10 @@ class HaloFindingMethod:
2727
dataset and returns a new dataset that is the loaded halo finder output.
2828
"""
2929

30-
def __init__(self, function, args=None, kwargs=None):
30+
def __init__(self, function, *args, **kwargs):
3131
self.function = function
3232
self.args = args
33-
if self.args is None:
34-
self.args = []
3533
self.kwargs = kwargs
36-
if self.kwargs is None:
37-
self.kwargs = {}
3834

3935
def __call__(self, hc):
4036
return self.function(hc, *self.args, **self.kwargs)

0 commit comments

Comments
 (0)