notebooks/shortcomings/graphviz-filter-out-vars.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "9e3faf70",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import pymc as pm\n",
    "from IPython.display import display\n",
    "from pymc.model_graph import ModelGraph\n",
    "\n",
    "from aesara.tensor.random.basic import RandomVariable\n",
    "from aesara.graph.basic import *\n",
    "import aesara\n",
    "\n",
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "519924e6",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 3.0.0 (0)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"200pt\" height=\"483pt\"\n",
       " viewBox=\"0.00 0.00 200.22 482.86\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 478.86)\">\n",
       "<polygon fill=\"white\" stroke=\"transparent\" points=\"-4,4 -4,-478.86 196.22,-478.86 196.22,4 -4,4\"/>\n",
       "<!-- pred -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>pred</title>\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"91,-252.95 0,-252.95 0,-199.95 91,-199.95 91,-252.95\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-237.75\" font-family=\"Times,serif\" font-size=\"14.00\">pred</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-222.75\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-207.75\" font-family=\"Times,serif\" font-size=\"14.00\">Deterministic</text>\n",
       "</g>\n",
       "<!-- L -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>L</title>\n",
       "<ellipse fill=\"lightgrey\" stroke=\"black\" cx=\"45.5\" cy=\"-126.48\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-137.78\" font-family=\"Times,serif\" font-size=\"14.00\">L</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-122.78\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-107.78\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- pred&#45;&gt;L -->\n",
       "<g id=\"edge7\" class=\"edge\">\n",
       "<title>pred&#45;&gt;L</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M45.5,-199.71C45.5,-191.89 45.5,-183.03 45.5,-174.27\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49,-174.1 45.5,-164.1 42,-174.1 49,-174.1\"/>\n",
       "</g>\n",
       "<!-- intermediate -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>intermediate</title>\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"91,-352.93 0,-352.93 0,-299.93 91,-299.93 91,-352.93\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-337.73\" font-family=\"Times,serif\" font-size=\"14.00\">intermediate</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-322.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-307.73\" font-family=\"Times,serif\" font-size=\"14.00\">Deterministic</text>\n",
       "</g>\n",
       "<!-- intermediate&#45;&gt;pred -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>intermediate&#45;&gt;pred</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M45.5,-299.69C45.5,-288.55 45.5,-275.32 45.5,-263.25\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49,-263.02 45.5,-253.02 42,-263.02 49,-263.02\"/>\n",
       "</g>\n",
       "<!-- obs -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>obs</title>\n",
       "<path fill=\"lightgrey\" stroke=\"black\" d=\"M78.5,-53C78.5,-53 12.5,-53 12.5,-53 6.5,-53 0.5,-47 0.5,-41 0.5,-41 0.5,-12 0.5,-12 0.5,-6 6.5,0 12.5,0 12.5,0 78.5,0 78.5,0 84.5,0 90.5,-6 90.5,-12 90.5,-12 90.5,-41 90.5,-41 90.5,-47 84.5,-53 78.5,-53\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-37.8\" font-family=\"Times,serif\" font-size=\"14.00\">obs</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-22.8\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-7.8\" font-family=\"Times,serif\" font-size=\"14.00\">ConstantData</text>\n",
       "</g>\n",
       "<!-- L&#45;&gt;obs -->\n",
       "<g id=\"edge6\" class=\"edge\">\n",
       "<title>L&#45;&gt;obs</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M45.5,-88.99C45.5,-80.58 45.5,-71.63 45.5,-63.25\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49,-63.01 45.5,-53.01 42,-63.01 49,-63.01\"/>\n",
       "</g>\n",
       "<!-- c -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>c</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"150.5\" cy=\"-326.43\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"150.5\" y=\"-337.73\" font-family=\"Times,serif\" font-size=\"14.00\">c</text>\n",
       "<text text-anchor=\"middle\" x=\"150.5\" y=\"-322.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"150.5\" y=\"-307.73\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- a -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>a</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"46.5\" cy=\"-437.38\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"46.5\" y=\"-448.68\" font-family=\"Times,serif\" font-size=\"14.00\">a</text>\n",
       "<text text-anchor=\"middle\" x=\"46.5\" y=\"-433.68\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"46.5\" y=\"-418.68\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- a&#45;&gt;intermediate -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>a&#45;&gt;intermediate</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M46.16,-399.8C46.06,-388.06 45.94,-375.08 45.83,-363.38\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49.33,-363.05 45.74,-353.08 42.33,-363.11 49.33,-363.05\"/>\n",
       "</g>\n",
       "<!-- a&#45;&gt;c -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>a&#45;&gt;c</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M73.28,-408.33C86.51,-394.47 102.6,-377.61 116.66,-362.88\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"119.44,-365.04 123.81,-355.39 114.37,-360.21 119.44,-365.04\"/>\n",
       "</g>\n",
       "<!-- b -->\n",
       "<g id=\"node7\" class=\"node\">\n",
       "<title>b</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"149.5\" cy=\"-437.38\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"149.5\" y=\"-448.68\" font-family=\"Times,serif\" font-size=\"14.00\">b</text>\n",
       "<text text-anchor=\"middle\" x=\"149.5\" y=\"-433.68\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"149.5\" y=\"-418.68\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- b&#45;&gt;intermediate -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>b&#45;&gt;intermediate</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M122.72,-408.33C108.75,-393.7 91.59,-375.71 76.99,-360.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"79.49,-357.97 70.05,-353.15 74.43,-362.8 79.49,-357.97\"/>\n",
       "</g>\n",
       "<!-- b&#45;&gt;c -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>b&#45;&gt;c</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M149.84,-399.8C149.91,-391.63 149.99,-382.85 150.07,-374.32\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"153.57,-374.13 150.16,-364.1 146.57,-374.07 153.57,-374.13\"/>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x149490bb0>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "with pm.Model() as pmodel2:\n",
    "    a = pm.Normal(\"a\")\n",
    "    b = pm.Normal(\"b\")\n",
    "    c = pm.Normal(\"c\", a * b)\n",
    "    intermediate = pm.Deterministic(\"intermediate\", a + b)\n",
    "    pred = pm.Deterministic(\"pred\", intermediate * 3)\n",
    "\n",
    "    obs = pm.ConstantData(\"obs\", 1.75)\n",
    "\n",
    "    L = pm.Normal(\"L\", mu=1 + 0.5 * pred, observed=obs)\n",
    "\n",
    "display(pm.model_to_graphviz(pmodel2))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "0085522c",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 3.0.0 (0)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"192pt\" height=\"194pt\"\n",
       " viewBox=\"0.00 0.00 192.44 193.91\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 189.91)\">\n",
       "<polygon fill=\"white\" stroke=\"transparent\" points=\"-4,4 -4,-189.91 188.44,-189.91 188.44,4 -4,4\"/>\n",
       "<!-- a -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>a</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"41.72\" cy=\"-148.43\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"41.72\" y=\"-159.73\" font-family=\"Times,serif\" font-size=\"14.00\">a</text>\n",
       "<text text-anchor=\"middle\" x=\"41.72\" y=\"-144.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"41.72\" y=\"-129.73\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- c -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>c</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"91.72\" cy=\"-37.48\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-48.78\" font-family=\"Times,serif\" font-size=\"14.00\">c</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-33.78\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-18.78\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- a&#45;&gt;c -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>a&#45;&gt;c</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M57.27,-113.55C61.92,-103.41 67.09,-92.14 71.98,-81.49\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"75.19,-82.88 76.18,-72.33 68.83,-79.96 75.19,-82.88\"/>\n",
       "</g>\n",
       "<!-- b -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>b</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"142.72\" cy=\"-148.43\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"142.72\" y=\"-159.73\" font-family=\"Times,serif\" font-size=\"14.00\">b</text>\n",
       "<text text-anchor=\"middle\" x=\"142.72\" y=\"-144.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"142.72\" y=\"-129.73\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- b&#45;&gt;c -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>b&#45;&gt;c</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M126.86,-113.55C122.11,-103.41 116.84,-92.14 111.85,-81.49\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"114.98,-79.91 107.57,-72.33 108.64,-82.87 114.98,-79.91\"/>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x14922ca00>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "display(pm.model_to_graphviz(pmodel2, var_names=[\"c\"]))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "14dfac46",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 3.0.0 (0)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"192pt\" height=\"461pt\"\n",
       " viewBox=\"0.00 0.00 192.44 460.91\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 456.91)\">\n",
       "<polygon fill=\"white\" stroke=\"transparent\" points=\"-4,4 -4,-456.91 188.44,-456.91 188.44,4 -4,4\"/>\n",
       "<!-- pred -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>pred</title>\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"137.22,-252.95 46.22,-252.95 46.22,-199.95 137.22,-199.95 137.22,-252.95\"/>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-237.75\" font-family=\"Times,serif\" font-size=\"14.00\">pred</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-222.75\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-207.75\" font-family=\"Times,serif\" font-size=\"14.00\">Deterministic</text>\n",
       "</g>\n",
       "<!-- L -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>L</title>\n",
       "<ellipse fill=\"lightgrey\" stroke=\"black\" cx=\"91.72\" cy=\"-126.48\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-137.78\" font-family=\"Times,serif\" font-size=\"14.00\">L</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-122.78\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-107.78\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- pred&#45;&gt;L -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>pred&#45;&gt;L</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M91.72,-199.71C91.72,-191.89 91.72,-183.03 91.72,-174.27\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"95.22,-174.1 91.72,-164.1 88.22,-174.1 95.22,-174.1\"/>\n",
       "</g>\n",
       "<!-- intermediate -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>intermediate</title>\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"137.22,-341.95 46.22,-341.95 46.22,-288.95 137.22,-288.95 137.22,-341.95\"/>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-326.75\" font-family=\"Times,serif\" font-size=\"14.00\">intermediate</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-311.75\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-296.75\" font-family=\"Times,serif\" font-size=\"14.00\">Deterministic</text>\n",
       "</g>\n",
       "<!-- intermediate&#45;&gt;pred -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>intermediate&#45;&gt;pred</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M91.72,-288.82C91.72,-280.84 91.72,-271.89 91.72,-263.34\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"95.22,-263.2 91.72,-253.2 88.22,-263.2 95.22,-263.2\"/>\n",
       "</g>\n",
       "<!-- obs -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>obs</title>\n",
       "<path fill=\"lightgrey\" stroke=\"black\" d=\"M124.72,-53C124.72,-53 58.72,-53 58.72,-53 52.72,-53 46.72,-47 46.72,-41 46.72,-41 46.72,-12 46.72,-12 46.72,-6 52.72,0 58.72,0 58.72,0 124.72,0 124.72,0 130.72,0 136.72,-6 136.72,-12 136.72,-12 136.72,-41 136.72,-41 136.72,-47 130.72,-53 124.72,-53\"/>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-37.8\" font-family=\"Times,serif\" font-size=\"14.00\">obs</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-22.8\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"91.72\" y=\"-7.8\" font-family=\"Times,serif\" font-size=\"14.00\">ConstantData</text>\n",
       "</g>\n",
       "<!-- L&#45;&gt;obs -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>L&#45;&gt;obs</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M91.72,-88.99C91.72,-80.58 91.72,-71.63 91.72,-63.25\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"95.22,-63.01 91.72,-53.01 88.22,-63.01 95.22,-63.01\"/>\n",
       "</g>\n",
       "<!-- a -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>a</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"41.72\" cy=\"-415.43\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"41.72\" y=\"-426.73\" font-family=\"Times,serif\" font-size=\"14.00\">a</text>\n",
       "<text text-anchor=\"middle\" x=\"41.72\" y=\"-411.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"41.72\" y=\"-396.73\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- a&#45;&gt;intermediate -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>a&#45;&gt;intermediate</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M58.79,-380.99C63.72,-371.32 69.11,-360.76 74.07,-351.03\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"77.25,-352.51 78.67,-342.01 71.01,-349.33 77.25,-352.51\"/>\n",
       "</g>\n",
       "<!-- b -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>b</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"142.72\" cy=\"-415.43\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"142.72\" y=\"-426.73\" font-family=\"Times,serif\" font-size=\"14.00\">b</text>\n",
       "<text text-anchor=\"middle\" x=\"142.72\" y=\"-411.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"142.72\" y=\"-396.73\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- b&#45;&gt;intermediate -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>b&#45;&gt;intermediate</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M125.45,-381.26C120.42,-371.58 114.9,-360.98 109.81,-351.21\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"112.82,-349.41 105.1,-342.16 106.61,-352.64 112.82,-349.41\"/>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x14966f700>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "display(pm.model_to_graphviz(pmodel2, var_names=[\"L\"]))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "5a234320",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n",
       "<!DOCTYPE svg PUBLIC \"-//W3C//DTD SVG 1.1//EN\"\n",
       " \"http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd\">\n",
       "<!-- Generated by graphviz version 3.0.0 (0)\n",
       " -->\n",
       "<!-- Pages: 1 -->\n",
       "<svg width=\"200pt\" height=\"483pt\"\n",
       " viewBox=\"0.00 0.00 200.22 482.86\" xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 478.86)\">\n",
       "<polygon fill=\"white\" stroke=\"transparent\" points=\"-4,4 -4,-478.86 196.22,-478.86 196.22,4 -4,4\"/>\n",
       "<!-- pred -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>pred</title>\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"91,-252.95 0,-252.95 0,-199.95 91,-199.95 91,-252.95\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-237.75\" font-family=\"Times,serif\" font-size=\"14.00\">pred</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-222.75\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-207.75\" font-family=\"Times,serif\" font-size=\"14.00\">Deterministic</text>\n",
       "</g>\n",
       "<!-- L -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>L</title>\n",
       "<ellipse fill=\"lightgrey\" stroke=\"black\" cx=\"45.5\" cy=\"-126.48\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-137.78\" font-family=\"Times,serif\" font-size=\"14.00\">L</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-122.78\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-107.78\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- pred&#45;&gt;L -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>pred&#45;&gt;L</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M45.5,-199.71C45.5,-191.89 45.5,-183.03 45.5,-174.27\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49,-174.1 45.5,-164.1 42,-174.1 49,-174.1\"/>\n",
       "</g>\n",
       "<!-- intermediate -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>intermediate</title>\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"91,-352.93 0,-352.93 0,-299.93 91,-299.93 91,-352.93\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-337.73\" font-family=\"Times,serif\" font-size=\"14.00\">intermediate</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-322.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-307.73\" font-family=\"Times,serif\" font-size=\"14.00\">Deterministic</text>\n",
       "</g>\n",
       "<!-- intermediate&#45;&gt;pred -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>intermediate&#45;&gt;pred</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M45.5,-299.69C45.5,-288.55 45.5,-275.32 45.5,-263.25\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49,-263.02 45.5,-253.02 42,-263.02 49,-263.02\"/>\n",
       "</g>\n",
       "<!-- obs -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>obs</title>\n",
       "<path fill=\"lightgrey\" stroke=\"black\" d=\"M78.5,-53C78.5,-53 12.5,-53 12.5,-53 6.5,-53 0.5,-47 0.5,-41 0.5,-41 0.5,-12 0.5,-12 0.5,-6 6.5,0 12.5,0 12.5,0 78.5,0 78.5,0 84.5,0 90.5,-6 90.5,-12 90.5,-12 90.5,-41 90.5,-41 90.5,-47 84.5,-53 78.5,-53\"/>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-37.8\" font-family=\"Times,serif\" font-size=\"14.00\">obs</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-22.8\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"45.5\" y=\"-7.8\" font-family=\"Times,serif\" font-size=\"14.00\">ConstantData</text>\n",
       "</g>\n",
       "<!-- L&#45;&gt;obs -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>L&#45;&gt;obs</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M45.5,-88.99C45.5,-80.58 45.5,-71.63 45.5,-63.25\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49,-63.01 45.5,-53.01 42,-63.01 49,-63.01\"/>\n",
       "</g>\n",
       "<!-- c -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>c</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"150.5\" cy=\"-326.43\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"150.5\" y=\"-337.73\" font-family=\"Times,serif\" font-size=\"14.00\">c</text>\n",
       "<text text-anchor=\"middle\" x=\"150.5\" y=\"-322.73\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"150.5\" y=\"-307.73\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- a -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>a</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"46.5\" cy=\"-437.38\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"46.5\" y=\"-448.68\" font-family=\"Times,serif\" font-size=\"14.00\">a</text>\n",
       "<text text-anchor=\"middle\" x=\"46.5\" y=\"-433.68\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"46.5\" y=\"-418.68\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- a&#45;&gt;intermediate -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>a&#45;&gt;intermediate</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M46.16,-399.8C46.06,-388.06 45.94,-375.08 45.83,-363.38\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"49.33,-363.05 45.74,-353.08 42.33,-363.11 49.33,-363.05\"/>\n",
       "</g>\n",
       "<!-- a&#45;&gt;c -->\n",
       "<g id=\"edge7\" class=\"edge\">\n",
       "<title>a&#45;&gt;c</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M73.28,-408.33C86.51,-394.47 102.6,-377.61 116.66,-362.88\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"119.44,-365.04 123.81,-355.39 114.37,-360.21 119.44,-365.04\"/>\n",
       "</g>\n",
       "<!-- b -->\n",
       "<g id=\"node7\" class=\"node\">\n",
       "<title>b</title>\n",
       "<ellipse fill=\"none\" stroke=\"black\" cx=\"149.5\" cy=\"-437.38\" rx=\"41.94\" ry=\"37.45\"/>\n",
       "<text text-anchor=\"middle\" x=\"149.5\" y=\"-448.68\" font-family=\"Times,serif\" font-size=\"14.00\">b</text>\n",
       "<text text-anchor=\"middle\" x=\"149.5\" y=\"-433.68\" font-family=\"Times,serif\" font-size=\"14.00\">~</text>\n",
       "<text text-anchor=\"middle\" x=\"149.5\" y=\"-418.68\" font-family=\"Times,serif\" font-size=\"14.00\">Normal</text>\n",
       "</g>\n",
       "<!-- b&#45;&gt;intermediate -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>b&#45;&gt;intermediate</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M122.72,-408.33C108.75,-393.7 91.59,-375.71 76.99,-360.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"79.49,-357.97 70.05,-353.15 74.43,-362.8 79.49,-357.97\"/>\n",
       "</g>\n",
       "<!-- b&#45;&gt;c -->\n",
       "<g id=\"edge6\" class=\"edge\">\n",
       "<title>b&#45;&gt;c</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M149.84,-399.8C149.91,-391.63 149.99,-382.85 150.07,-374.32\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"153.57,-374.13 150.16,-364.1 146.57,-374.07 153.57,-374.13\"/>\n",
       "</g>\n",
       "</g>\n",
       "</svg>\n"
      ],
      "text/plain": [
       "<graphviz.graphs.Digraph at 0x1495bf610>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "display(pm.model_to_graphviz(pmodel2, var_names=[\"c\", \"L\"]))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "96c33784",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['obs', 'intermediate', 'pred', 'L', 'b', 'c', 'a']"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pm.model_graph.ModelGraph(pmodel2).vars_to_plot([\"c\", \"L\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "id": "2bff71d4",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'intermediate'}"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pm.model_graph.ModelGraph(pmodel2).get_parent_names(pred)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "id": "55de626a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<pymc.model_graph.ModelGraph at 0x149664520>"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ModelGraph(pmodel2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "e7e3e3d6",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[Elemwise{mul,no_inplace}.0,\n",
       " TensorConstant{3},\n",
       " intermediate ~ Deterministic(f(b, a)),\n",
       " Elemwise{add,no_inplace}.0,\n",
       " b ~ N(0, 1),\n",
       " a ~ N(0, 1)]"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "def _expand(x):\n",
    "    if x.owner and (not isinstance(x.owner.op, RandomVariable)):\n",
    "        return reversed(x.owner.inputs)\n",
    "\n",
    "list(walk(nodes=pred.owner.inputs, expand=_expand))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "eb6ed9b1",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "False"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from aesara.graph import Apply\n",
    "\n",
    "isinstance(None, Apply)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "367e57aa",
   "metadata": {},
   "outputs": [],
   "source": [
    "if isinstance(pred.owner.inputs[0], Apply):\n",
    "    print(\"hello\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "50c394dd",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'intermediate'"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pred.owner.inputs[0].owner.inputs[0].name"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "88047e74",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "set()"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ModelGraph(pmodel2).get_parent_names(a)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "88cdf82d",
   "metadata": {},
   "outputs": [],
   "source": [
    "if None:\n",
    "    print(1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "id": "e34c7b49",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'L'"
      ]
     },
     "execution_count": 15,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pm.util.get_var_name(L)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "77a96651",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "py39",
   "language": "python",
   "name": "py39"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}