Skip to content

Commit

Permalink
Merge pull request #183 from EvieQ01/patch-1
Browse files Browse the repository at this point in the history
Add node_names to GRaSP.py
  • Loading branch information
kunwuz authored Jun 24, 2024
2 parents 5de8ee6 + 52bf6f9 commit 8b7f591
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion causallearn/search/PermutationBased/GRaSP.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,7 @@ def grasp(
maxP: Optional[float] = None,
parameters: Optional[Dict[str, Any]] = None,
verbose: Optional[bool] = True,
node_names: Optional[List[str]] = None,
) -> GeneralGraph:
"""
Perform a greedy relaxation of the sparsest permutation (GRaSP) algorithm
Expand Down Expand Up @@ -185,7 +186,7 @@ def grasp(
raise Exception("Unknown function!")
score = localScoreClass

node_names = [("x%d" % i) for i in range(p)]
node_names = [("X%d" % (i + 1)) for i in range(p)] if node_names is None else node_names
nodes = []

for name in node_names:
Expand Down

0 comments on commit 8b7f591

Please sign in to comment.