[ENH] Add the ability to check the validity of a PAG

.circleci/config.yml

@@ -7,7 +7,7 @@ commands:
       - run:
           name: Check-skip
           command: |
-            if [ ! -d "sktree" ]; then
+            if [ ! -d "pywhy_graphs" ]; then
               echo "Build was not run due to skip, exiting job ${CIRCLE_JOB} for PR ${CIRCLE_PULL_REQUEST}."
               circleci-agent step halt;
             fi
@@ -54,7 +54,7 @@ commands:
             echo ${CI_PULL_REQUEST//*pull\//} | tee merge.txt
             if [[ $(cat merge.txt) != "" ]]; then
               echo "Merging $(cat merge.txt)";
-              git remote add upstream https://github.com/neurodata/scikit-tree.git;
+              git remote add upstream https://github.com/py-why/pywhy-graphs.git;
               git pull --ff-only upstream "refs/pull/$(cat merge.txt)/merge";
               git fetch upstream main;
             fi
@@ -64,7 +64,7 @@ jobs:
     docker:
       # CircleCI maintains a library of pre-built images
       # documented at https://circleci.com/doc/2.0/circleci-images/
-      - image: cimg/python:3.9
+      - image: cimg/python:3.11
     steps:
       - checkout
       - check-skip
@@ -96,18 +96,26 @@ jobs:
           name: Setup torch for pgmpy
           command: |
             sudo apt-get install nvidia-cuda-toolkit nvidia-cuda-toolkit-gcc
+      - run:
+          name: Install dodiscover
+          command: |
+            git clone https://github.com/py-why/dodiscover.git
+            cd dodiscover
+            python -m pip install .
       - run:
           name: Check installation
           command: |
             python -c "import pywhy_graphs;"
             python -c "import numpy; numpy.show_config()"
+            python -c "import dodiscover;"
             LIBGL_DEBUG=verbose python -c "import matplotlib.pyplot as plt; plt.figure()"
 
       # dowhy currently requires an older version of numpy
-      - run:
-          name: Temporary Hack for numpy
-          command: |
-            python -m pip install numpy==1.22.0
+      # - run:
+      #     name: Temporary Hack for numpy
+      #     command: |
+      #       python -m pip install numpy==1.22.0
+
 
       - run:
           name: Build documentation

.github/workflows/main.yml

@@ -117,11 +117,26 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           architecture: "x64"
+
+      - name: Setup torch for pgmpy
+        if: "matrix.os == 'ubuntu'"
+        shell: bash
+        run: |
+          sudo apt-get update
+          sudo apt-get install nvidia-cuda-toolkit nvidia-cuda-toolkit-gcc
+
       - name: Install packages via pip
         run: |
-          pip install --upgrade pip
-          pip install numpy scipy networkx statsmodels
-          pip install .[test]
+          python -m pip install --upgrade pip
+          python -m pip install numpy scipy networkx statsmodels
+          python -m pip install .[test]
+
+      - name: Install DoDiscover (main)
+        run: |
+          git clone https://github.com/py-why/dodiscover.git
+          cd dodiscover
+          python -m pip install .
+
       - name: Install Networkx (main)
         if: "matrix.networkx == 'main'"
         run: |
@@ -131,16 +146,10 @@ jobs:
           pip install .[default]
           # pip install --progress-bar off git+https://github.com/networkx/networkx
 
-      - name: Setup torch for pgmpy
-        if: "matrix.os == 'ubuntu'"
-        shell: bash
-        run: |
-          sudo apt-get update
-          sudo apt-get install nvidia-cuda-toolkit nvidia-cuda-toolkit-gcc
-
       - name: Run pytest # headless via Xvfb on linux
         run: |
           pytest --cov pywhy_graphs ./pywhy_graphs
+
       - name: Upload coverage stats to codecov
         if: ${{ matrix.os == 'ubuntu' && matrix.python-version == '3.11' && matrix.networkx == 'stable' }}
         uses: codecov/codecov-action@v4

doc/whats_new/v0.2.rst

@@ -32,6 +32,7 @@ Changelog
 - |Feature| Implement a suite of functions for finding and checking semi-directed paths on a mixed-edge graph, by `Adam Li`_ (:pr:`101`)
 - |Feature| Implement functions for converting between a DAG and PDAG and CPDAG for generating consistent extensions of a CPDAG for example. These functions are :func:`pywhy_graphs.algorithms.pdag_to_cpdag`, :func:`pywhy_graphs.algorithms.pdag_to_dag` and :func:`pywhy_graphs.algorithms.dag_to_cpdag`, by `Adam Li`_ (:pr:`102`)
 - |API| Remove poetry based setup, by `Adam Li`_ (:pr:`110`)
+- |Feature| Implement and test function to validate PAG, by `Aryan Roy`_ (:pr:`100`)
 
 Code and Documentation Contributors
 -----------------------------------

examples/intro/checking_validity_of_a_pag.py

@@ -0,0 +1,68 @@
+"""
+===========================
+On PAGs and their validity
+===========================
+
+A PAG or a Partial Ancestral Graph is a type of mixed edge
+graph that can represent, in a single graph, the causal relationship
+between several nodes as defined by an equivalence class of MAGs.
+
+
+PAGs model this relationship by displaying all common edge marks shared 
+by all members in the equivalence class and displaying circles for those marks
+that are not common.
+
+More details on inducing paths can be found at :footcite:`Zhang2008`.
+
+"""
+
+import pywhy_graphs
+from pywhy_graphs.viz import draw
+from pywhy_graphs import PAG
+
+try:
+    from dodiscover import FCI, make_context
+    from dodiscover.ci import Oracle
+    from dodiscover.constraint.utils import dummy_sample
+except ImportError as e:
+    raise ImportError("The 'dodiscover' package is required to convert a MAG to a PAG.")
+
+
+# %%
+# PAGs in pywhy-graphs
+# ---------------------------
+# Constructing a PAG in pywhy-graphs is an easy task since
+# the library provides a separate class for this purpose.
+# True to the definition of PAGs, the class can contain
+# directed edges, bidirected edges, undirected edges and
+# cicle edges. To illustrate this, we construct an example PAG:
+
+pag = PAG()
+pag.add_edge("A", "B", pag.directed_edge_name)
+pag.add_edge("B", "A", pag.circle_edge_name)
+pag.add_edge("B", "C", pag.circle_edge_name)
+pag.add_edge("C", "B", pag.directed_edge_name)
+pag.add_edge("A", "D", pag.circle_edge_name)
+pag.add_edge("D", "A", pag.circle_edge_name)
+
+
+# Finally, the graph looks like this:
+dot_graph = draw(pag)
+dot_graph.render(outfile="new_pag.png", view=True)
+
+
+# %%
+# Validity of a PAG
+# ---------------------------
+# To check if the constructed PAG is a valid one, we
+# can simply do:
+
+
+# returns True
+print(pywhy_graphs.valid_pag(pag))
+
+
+# %%
+# References
+# ----------
+# .. footbibliography::