Skip to content

Commit 9651824

Browse files
committed
edit docstring
Signed-off-by: Oliver Schacht <[email protected]>
1 parent fba553f commit 9651824

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

causallearn/utils/cit.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ def CIT(data, method='fisherz', **kwargs):
2828
Parameters
2929
----------
3030
data: numpy.ndarray of shape (n_samples, n_features)
31-
method: str, in ["fisherz", "mv_fisherz", "mc_fisherz", "kci", "chisq", "gsq"]
32-
kwargs: placeholder for future arguments, or for KCI specific arguments now
31+
method: str, in ["fisherz", "mv_fisherz", "mc_fisherz", "kci", "rcit", "fastkci", "chisq", "gsq"]
32+
kwargs: placeholder for future arguments, or for KCI, FastKCI or RCIT specific arguments now
3333
TODO: utimately kwargs should be replaced by explicit named parameters.
3434
check https://github.com/cmu-phil/causal-learn/pull/62#discussion_r927239028
3535
'''
@@ -52,6 +52,7 @@ def CIT(data, method='fisherz', **kwargs):
5252
else:
5353
raise ValueError("Unknown method: {}".format(method))
5454

55+
5556
class CIT_Base(object):
5657
# Base class for CIT, contains basic operations for input check and caching, etc.
5758
def __init__(self, data, cache_path=None, **kwargs):

0 commit comments

Comments
 (0)