fixed bug in efficient backdoor notebook

Signed-off-by: Amit Sharma <amit_sharma@live.com>

Author: amit-sharma

docs/source/example_notebooks/dowhy_efficient_backdoor_example.ipynb

@@ -62,7 +62,9 @@
    "outputs": [],
    "source": [
     "from dowhy.causal_graph import CausalGraph\n",
-    "from dowhy.causal_identifier import AutoIdentifier, BackdoorAdjustment, EstimandType"
+    "from dowhy.causal_identifier import AutoIdentifier, BackdoorAdjustment, EstimandType\n",
+    "from dowhy.graph import build_graph_from_str\n",
+    "from dowhy.utils.plotting import plot"
    ]
   },
   {
@@ -135,9 +137,7 @@
     "]\n",
     "treatment_name = \"warm-up\"\n",
     "outcome_name = \"injury\"\n",
-    "G = CausalGraph(\n",
-    "    graph=graph_str, treatment_name=treatment_name, outcome_name=outcome_name, observed_node_names=observed_node_names\n",
-    ")"
+    "G = build_graph_from_str(graph_str)"
    ]
   },
   {
@@ -153,7 +153,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "G.view_graph()"
+    "plot(G)"
    ]
   },
   {
@@ -184,7 +184,11 @@
     ")\n",
     "print(\n",
     "    ident_eff.identify_effect(\n",
-    "        graph=G, treatment_name=treatment_name, outcome_name=outcome_name, conditional_node_names=conditional_node_names\n",
+    "        graph=G, \n",
+    "        action_nodes=treatment_name, \n",
+    "        outcome_nodes=outcome_name,\n",
+    "        observed_nodes=observed_node_names,\n",
+    "        conditional_node_names=conditional_node_names\n",
     "    )\n",
     ")"
    ]
@@ -215,7 +219,11 @@
     ")\n",
     "print(\n",
     "    ident_minimal_eff.identify_effect(\n",
-    "        graph=G, treatment_name=treatment_name, outcome_name=outcome_name, conditional_node_names=conditional_node_names\n",
+    "        graph=G, \n",
+    "        action_nodes=treatment_name, \n",
+    "        outcome_nodes=outcome_name, \n",
+    "        observed_nodes=observed_node_names,\n",
+    "        conditional_node_names=conditional_node_names\n",
     "    )\n",
     ")"
    ]
@@ -239,7 +247,11 @@
     ")\n",
     "print(\n",
     "    ident_mincost_eff.identify_effect(\n",
-    "        graph=G, treatment_name=treatment_name, outcome_name=outcome_name, conditional_node_names=conditional_node_names\n",
+    "        graph=G, \n",
+    "        action_nodes=treatment_name, \n",
+    "        outcome_nodes=outcome_name,\n",
+    "        observed_nodes=observed_node_names,\n",
+    "        conditional_node_names=conditional_node_names\n",
     "    )\n",
     ")"
    ]
@@ -294,9 +306,7 @@
     "observed_node_names = [\"X\", \"Y\", \"Z1\", \"Z2\"]\n",
     "treatment_name = \"X\"\n",
     "outcome_name = \"Y\"\n",
-    "G = CausalGraph(\n",
-    "    graph=graph_str, treatment_name=treatment_name, outcome_name=outcome_name, observed_node_names=observed_node_names\n",
-    ")"
+    "G = build_graph_from_str(graph_str)"
    ]
   },
   {
@@ -317,7 +327,10 @@
     "    backdoor_adjustment=BackdoorAdjustment.BACKDOOR_EFFICIENT,\n",
     ")\n",
     "try:\n",
-    "    results_eff = ident_eff.identify_effect(graph=G, treatment_name=treatment_name, outcome_name=outcome_name)\n",
+    "    results_eff = ident_eff.identify_effect(graph=G, \n",
+    "                                            action_nodes=treatment_name, \n",
+    "                                            outcome_nodes=outcome_name,\n",
+    "                                            observed_nodes=observed_node_names)\n",
     "except ValueError as e:\n",
     "    print(e)"
    ]
@@ -335,8 +348,9 @@
     "print(\n",
     "    ident_minimal_eff.identify_effect(\n",
     "        graph=G,\n",
-    "        treatment_name=treatment_name,\n",
-    "        outcome_name=outcome_name,\n",
+    "        action_nodes=treatment_name,\n",
+    "        outcome_nodes=outcome_name,\n",
+    "        observed_nodes=observed_node_names\n",
     "    )\n",
     ")"
    ]
@@ -354,8 +368,9 @@
     "print(\n",
     "    ident_mincost_eff.identify_effect(\n",
     "        graph=G,\n",
-    "        treatment_name=treatment_name,\n",
-    "        outcome_name=outcome_name,\n",
+    "        action_nodes=treatment_name,\n",
+    "        outcome_nodes=outcome_name,\n",
+    "        observed_nodes=observed_node_names\n",
     "    )\n",
     ")"
    ]
@@ -391,9 +406,7 @@
     "observed_node_names = [\"X\", \"Y\"]\n",
     "treatment_name = \"X\"\n",
     "outcome_name = \"Y\"\n",
-    "G = CausalGraph(\n",
-    "    graph=graph_str, treatment_name=treatment_name, outcome_name=outcome_name, observed_node_names=observed_node_names\n",
-    ")"
+    "G = build_graph_from_str(graph_str)"
    ]
   },
   {
@@ -409,8 +422,9 @@
     "try:\n",
     "    results_eff = ident_eff.identify_effect(\n",
     "        graph=G,\n",
-    "        treatment_name=treatment_name,\n",
-    "        outcome_name=outcome_name,\n",
+    "        action_nodes=treatment_name,\n",
+    "        outcome_nodes=outcome_name,\n",
+    "        observed_nodes=observed_node_names\n",
     "    )\n",
     "except ValueError as e:\n",
     "    print(e)"
@@ -475,9 +489,7 @@
     "    (\"R\", {\"cost\": 2}),\n",
     "    (\"T\", {\"cost\": 1}),\n",
     "]\n",
-    "G = CausalGraph(\n",
-    "    graph=graph_str, treatment_name=treatment_name, outcome_name=outcome_name, observed_node_names=observed_node_names\n",
-    ")"
+    "G = build_graph_from_str(graph_str)"
    ]
   },
   {
@@ -504,7 +516,11 @@
     ")\n",
     "print(\n",
     "    ident_mincost_eff.identify_effect(\n",
-    "        graph=G, treatment_name=treatment_name, outcome_name=outcome_name, conditional_node_names=conditional_node_names\n",
+    "        graph=G, \n",
+    "        action_nodes=treatment_name, \n",
+    "        outcome_nodes=outcome_name, \n",
+    "        observed_nodes=observed_node_names,\n",
+    "        conditional_node_names=conditional_node_names\n",
     "    )\n",
     ")"
    ]
@@ -528,22 +544,19 @@
     ")\n",
     "print(\n",
     "    ident_minimal_eff.identify_effect(\n",
-    "        graph=G, treatment_name=treatment_name, outcome_name=outcome_name, conditional_node_names=conditional_node_names\n",
+    "        graph=G, \n",
+    "        action_nodes=treatment_name,\n",
+    "        outcome_nodes=outcome_name, \n",
+    "        observed_nodes=observed_node_names,\n",
+    "        conditional_node_names=conditional_node_names\n",
     "    )\n",
     ")"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3.8.10 ('dowhy-_zBapv7Q-py3.8')",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
    "name": "python3"
   },