Skip to content

Commit ef3bff9

Browse files
committed
remove further causalgraphicalmodel code
1 parent 0df6a97 commit ef3bff9

File tree

1 file changed

+8
-17
lines changed

1 file changed

+8
-17
lines changed

notebooks/05_the_many_variables_and_the_spurious_waffles.ipynb

Lines changed: 8 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -1690,28 +1690,19 @@
16901690
"metadata": {},
16911691
"outputs": [],
16921692
"source": [
1693-
"dag5_7 = CausalGraphicalModel(\n",
1694-
" nodes=[\"M\", \"K\", \"N\"], edges=[(\"M\", \"K\"), (\"N\", \"K\"), (\"M\", \"N\")]\n",
1695-
")\n",
1693+
"dag5_7 = nx.DiGraph()\n",
1694+
"dag5_7.add_edges_from([(\"M\", \"K\"), (\"N\", \"K\"), (\"M\", \"N\")])\n",
16961695
"coordinates = {\"M\": (0, 0.5), \"K\": (1, 1), \"N\": (2, 0.5)}\n",
1697-
"nodes = list(dag5_7.dag.nodes.keys())\n",
1698-
"edges = list(dag5_7.dag.edges.keys())\n",
16991696
"MElist = []\n",
17001697
"for i in range(2):\n",
17011698
" for j in range(2):\n",
17021699
" for k in range(2):\n",
1703-
" try:\n",
1704-
" new_dag = CausalGraphicalModel(\n",
1705-
" nodes=nodes,\n",
1706-
" edges=[\n",
1707-
" edges[0] if i == 0 else edges[0][::-1],\n",
1708-
" edges[1] if j == 0 else edges[1][::-1],\n",
1709-
" edges[2] if k == 0 else edges[2][::-1],\n",
1710-
" ],\n",
1711-
" )\n",
1712-
" MElist.append(new_dag)\n",
1713-
" except:\n",
1714-
" pass"
1700+
" new_dag = nx.DiGraph()\n",
1701+
" new_dag.add_edges_from(\n",
1702+
" [edge[::-1] if flip else edge for edge, flip in zip(dag5_7.edges, (i, j, k))]\n",
1703+
" )\n",
1704+
" if not list(nx.simple_cycles(new_dag)):\n",
1705+
" MElist.append(new_dag)"
17151706
]
17161707
},
17171708
{

0 commit comments

Comments
 (0)