Skip to content

Commit 4898e8a

Browse files
committed
feat: optionally disable progress bar in fci
1 parent f511954 commit 4898e8a

File tree

1 file changed

+2
-2
lines changed
  • causallearn/search/ConstraintBased

1 file changed

+2
-2
lines changed

causallearn/search/ConstraintBased/FCI.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -693,7 +693,7 @@ def _contains_all(set_a: Set[Node], set_b: Set[Node]):
693693

694694

695695
def fci(dataset: ndarray, independence_test_method: str=fisherz, alpha: float = 0.05, depth: int = -1,
696-
max_path_length: int = -1, verbose: bool = False, background_knowledge: BackgroundKnowledge | None = None,
696+
max_path_length: int = -1, verbose: bool = False, background_knowledge: BackgroundKnowledge | None = None, show_progress: bool = True,
697697
**kwargs) -> Tuple[Graph, List[Edge]]:
698698
"""
699699
Perform Fast Causal Inference (FCI) algorithm for causal discovery
@@ -755,7 +755,7 @@ def fci(dataset: ndarray, independence_test_method: str=fisherz, alpha: float =
755755

756756
# FAS (“Fast Adjacency Search”) is the adjacency search of the PC algorithm, used as a first step for the FCI algorithm.
757757
graph, sep_sets, test_results = fas(dataset, nodes, independence_test_method=independence_test_method, alpha=alpha,
758-
knowledge=background_knowledge, depth=depth, verbose=verbose)
758+
knowledge=background_knowledge, depth=depth, verbose=verbose, show_progress=show_progress)
759759

760760
reorientAllWith(graph, Endpoint.CIRCLE)
761761

0 commit comments

Comments
 (0)