examples/pangeo-vorticity.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "f5a954d5",
   "metadata": {},
   "source": [
    "# Pangeo Vorticity Workload\n",
    "\n",
    "This is a notebook for exploring a simplified version of the example in https://github.com/pangeo-data/distributed-array-examples/issues/1.\n",
    "\n",
    "In particular, it is the code from https://github.com/coiled/coiled-runtime/issues/174"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "8d745a7b",
   "metadata": {},
   "outputs": [],
   "source": [
    "import cubed.array_api as xp\n",
    "import cubed.random\n",
    "from cubed.extensions.tqdm import TqdmProgressBar"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "1bff01e3",
   "metadata": {},
   "outputs": [],
   "source": [
    "a = cubed.random.random((1000, 900, 800), chunks=100)\n",
    "b = cubed.random.random((1000, 900, 800), chunks=100)\n",
    "x = cubed.random.random((900, 800), chunks=100)\n",
    "y = cubed.random.random((900, 800), chunks=100)\n",
    "\n",
    "result = a[1:] * x + b[1:] * y\n",
    "result = xp.mean(result)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "f8ca9389",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "image/svg+xml": [
       "<svg xmlns=\"http://www.w3.org/2000/svg\" xmlns:xlink=\"http://www.w3.org/1999/xlink\" width=\"971pt\" height=\"578pt\" viewBox=\"0.00 0.00 971.00 578.00\">\n",
       "<g id=\"graph0\" class=\"graph\" transform=\"scale(1 1) rotate(0) translate(4 574)\">\n",
       "<polygon fill=\"white\" stroke=\"transparent\" points=\"-4,4 -4,-574 967,-574 967,4 -4,4\"/>\n",
       "<text text-anchor=\"middle\" x=\"75.5\" y=\"-18\" font-family=\"Times,serif\" font-size=\"10.00\">num tasks: 5192</text>\n",
       "<text text-anchor=\"middle\" x=\"75.5\" y=\"-7\" font-family=\"Times,serif\" font-size=\"10.00\">max required memory: 100.0 MB</text>\n",
       "<!-- array&#45;017 -->\n",
       "<g id=\"node1\" class=\"node\">\n",
       "<title>array-017</title>\n",
       "<g id=\"a_node1\"><a xlink:title=\"shape: (999, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; empty -&gt; full\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"692,-426 618,-426 618,-390 692,-390 692,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"655\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-017</text>\n",
       "<text text-anchor=\"middle\" x=\"655\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;020 -->\n",
       "<g id=\"node8\" class=\"node\">\n",
       "<title>array-020</title>\n",
       "<g id=\"a_node8\"><a xlink:title=\"shape: (999, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "task memory: 32.0 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"752,-354 658,-354 658,-318 752,-318 752,-354\"/>\n",
       "<text text-anchor=\"middle\" x=\"705\" y=\"-339\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-020</text>\n",
       "<text text-anchor=\"middle\" x=\"705\" y=\"-328\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;017&#45;&gt;array&#45;020 -->\n",
       "<g id=\"edge1\" class=\"edge\">\n",
       "<title>array-017-&gt;array-020</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M667.36,-389.7C673.23,-381.47 680.37,-371.48 686.84,-362.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"689.82,-364.28 692.78,-354.1 684.12,-360.21 689.82,-364.28\"/>\n",
       "</g>\n",
       "<!-- array&#45;018 -->\n",
       "<g id=\"node2\" class=\"node\">\n",
       "<title>array-018</title>\n",
       "<g id=\"a_node2\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (10, 9, 8)\n",
       "dtype: int32\n",
       "chunk memory: 2.9 KB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; map_blocks -&gt; asarray\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"792,-498 718,-498 718,-462 792,-462 792,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"755\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-018</text>\n",
       "<text text-anchor=\"middle\" x=\"755\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;019 -->\n",
       "<g id=\"node3\" class=\"node\">\n",
       "<title>array-019</title>\n",
       "<g id=\"a_node3\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (1, 1, 1)\n",
       "dtype: int32\n",
       "chunk memory: 4 bytes\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; map_blocks -&gt; rechunk\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"799.5,-426 710.5,-426 710.5,-390 799.5,-390 799.5,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"755\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-019</text>\n",
       "<text text-anchor=\"middle\" x=\"755\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;018&#45;&gt;array&#45;019 -->\n",
       "<g id=\"edge2\" class=\"edge\">\n",
       "<title>array-018-&gt;array-019</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M755,-461.7C755,-453.98 755,-444.71 755,-436.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"758.5,-436.1 755,-426.1 751.5,-436.1 758.5,-436.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;019&#45;&gt;array&#45;020 -->\n",
       "<g id=\"edge3\" class=\"edge\">\n",
       "<title>array-019-&gt;array-020</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M742.64,-389.7C736.77,-381.47 729.63,-371.48 723.16,-362.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"725.88,-360.21 717.22,-354.1 720.18,-364.28 725.88,-360.21\"/>\n",
       "</g>\n",
       "<!-- array&#45;001 -->\n",
       "<g id=\"node4\" class=\"node\">\n",
       "<title>array-001</title>\n",
       "<g id=\"a_node4\"><a xlink:title=\"shape: (1000, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; empty -&gt; full\n",
       "line: 1 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"876.5,-498 817.5,-498 817.5,-462 876.5,-462 876.5,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"847\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-001</text>\n",
       "<text text-anchor=\"middle\" x=\"847\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;004 -->\n",
       "<g id=\"node7\" class=\"node\">\n",
       "<title>array-004</title>\n",
       "<g id=\"a_node7\"><a xlink:title=\"shape: (1000, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "task memory: 32.0 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 1 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"890.5,-426 817.5,-426 817.5,-390 890.5,-390 890.5,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"854\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-004</text>\n",
       "<text text-anchor=\"middle\" x=\"854\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;001&#45;&gt;array&#45;004 -->\n",
       "<g id=\"edge4\" class=\"edge\">\n",
       "<title>array-001-&gt;array-004</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M848.73,-461.7C849.5,-453.98 850.43,-444.71 851.29,-436.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"854.78,-436.4 852.29,-426.1 847.81,-435.71 854.78,-436.4\"/>\n",
       "</g>\n",
       "<!-- array&#45;002 -->\n",
       "<g id=\"node5\" class=\"node\">\n",
       "<title>array-002</title>\n",
       "<g id=\"a_node5\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (10, 9, 8)\n",
       "dtype: int32\n",
       "chunk memory: 2.9 KB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; asarray\n",
       "line: 1 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"958.5,-570 899.5,-570 899.5,-534 958.5,-534 958.5,-570\"/>\n",
       "<text text-anchor=\"middle\" x=\"929\" y=\"-555\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-002</text>\n",
       "<text text-anchor=\"middle\" x=\"929\" y=\"-544\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;003 -->\n",
       "<g id=\"node6\" class=\"node\">\n",
       "<title>array-003</title>\n",
       "<g id=\"a_node6\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (1, 1, 1)\n",
       "dtype: int32\n",
       "chunk memory: 4 bytes\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; rechunk\n",
       "line: 1 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"963,-498 895,-498 895,-462 963,-462 963,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"929\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-003</text>\n",
       "<text text-anchor=\"middle\" x=\"929\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;002&#45;&gt;array&#45;003 -->\n",
       "<g id=\"edge5\" class=\"edge\">\n",
       "<title>array-002-&gt;array-003</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M929,-533.7C929,-525.98 929,-516.71 929,-508.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"932.5,-508.1 929,-498.1 925.5,-508.1 932.5,-508.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;003&#45;&gt;array&#45;004 -->\n",
       "<g id=\"edge6\" class=\"edge\">\n",
       "<title>array-003-&gt;array-004</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M910.46,-461.7C901.2,-453.05 889.84,-442.45 879.74,-433.03\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"882.02,-430.37 872.33,-426.1 877.25,-435.49 882.02,-430.37\"/>\n",
       "</g>\n",
       "<!-- array&#45;004&#45;&gt;array&#45;020 -->\n",
       "<g id=\"edge7\" class=\"edge\">\n",
       "<title>array-004-&gt;array-020</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M817.55,-389.88C797.36,-380.39 772.08,-368.51 750.59,-358.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"751.87,-355.15 741.33,-354.07 748.89,-361.49 751.87,-355.15\"/>\n",
       "</g>\n",
       "<!-- array&#45;021 -->\n",
       "<g id=\"node13\" class=\"node\">\n",
       "<title>array-021</title>\n",
       "<g id=\"a_node13\"><a xlink:title=\"shape: (999, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "task memory: 32.2 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; __mul__ -&gt; elemwise -&gt; blockwise\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"541.5,-282 464.5,-282 464.5,-246 541.5,-246 541.5,-282\"/>\n",
       "<text text-anchor=\"middle\" x=\"503\" y=\"-267\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-021</text>\n",
       "<text text-anchor=\"middle\" x=\"503\" y=\"-256\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__mul__ (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;020&#45;&gt;array&#45;021 -->\n",
       "<g id=\"edge8\" class=\"edge\">\n",
       "<title>array-020-&gt;array-021</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M657.89,-318.68C625.88,-307.58 583.65,-292.95 551.38,-281.76\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"552.3,-278.38 541.7,-278.41 550,-284.99 552.3,-278.38\"/>\n",
       "</g>\n",
       "<!-- array&#45;009 -->\n",
       "<g id=\"node9\" class=\"node\">\n",
       "<title>array-009</title>\n",
       "<g id=\"a_node9\"><a xlink:title=\"shape: (900, 800)\n",
       "chunks: (100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 80.0 KB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; empty -&gt; full\n",
       "line: 3 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"513.5,-426 454.5,-426 454.5,-390 513.5,-390 513.5,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"484\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-009</text>\n",
       "<text text-anchor=\"middle\" x=\"484\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;012 -->\n",
       "<g id=\"node12\" class=\"node\">\n",
       "<title>array-012</title>\n",
       "<g id=\"a_node12\"><a xlink:title=\"shape: (900, 800)\n",
       "chunks: (100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 80.0 KB\n",
       "\n",
       "task memory: 320.0 KB\n",
       "tasks: 72\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 3 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"539.5,-354 466.5,-354 466.5,-318 539.5,-318 539.5,-354\"/>\n",
       "<text text-anchor=\"middle\" x=\"503\" y=\"-339\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-012</text>\n",
       "<text text-anchor=\"middle\" x=\"503\" y=\"-328\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;009&#45;&gt;array&#45;012 -->\n",
       "<g id=\"edge9\" class=\"edge\">\n",
       "<title>array-009-&gt;array-012</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M488.7,-389.7C490.81,-381.9 493.36,-372.51 495.72,-363.83\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"499.12,-364.67 498.36,-354.1 492.36,-362.84 499.12,-364.67\"/>\n",
       "</g>\n",
       "<!-- array&#45;010 -->\n",
       "<g id=\"node10\" class=\"node\">\n",
       "<title>array-010</title>\n",
       "<g id=\"a_node10\"><a xlink:title=\"shape: (9, 8)\n",
       "chunks: (9, 8)\n",
       "dtype: int32\n",
       "chunk memory: 288 bytes\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; asarray\n",
       "line: 3 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"595.5,-498 536.5,-498 536.5,-462 595.5,-462 595.5,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"566\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-010</text>\n",
       "<text text-anchor=\"middle\" x=\"566\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;011 -->\n",
       "<g id=\"node11\" class=\"node\">\n",
       "<title>array-011</title>\n",
       "<g id=\"a_node11\"><a xlink:title=\"shape: (9, 8)\n",
       "chunks: (1, 1)\n",
       "dtype: int32\n",
       "chunk memory: 4 bytes\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; rechunk\n",
       "line: 3 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"600,-426 532,-426 532,-390 600,-390 600,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"566\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-011</text>\n",
       "<text text-anchor=\"middle\" x=\"566\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;010&#45;&gt;array&#45;011 -->\n",
       "<g id=\"edge10\" class=\"edge\">\n",
       "<title>array-010-&gt;array-011</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M566,-461.7C566,-453.98 566,-444.71 566,-436.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"569.5,-436.1 566,-426.1 562.5,-436.1 569.5,-436.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;011&#45;&gt;array&#45;012 -->\n",
       "<g id=\"edge11\" class=\"edge\">\n",
       "<title>array-011-&gt;array-012</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M550.43,-389.7C542.8,-381.22 533.48,-370.86 525.12,-361.58\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"527.69,-359.2 518.39,-354.1 522.48,-363.88 527.69,-359.2\"/>\n",
       "</g>\n",
       "<!-- array&#45;012&#45;&gt;array&#45;021 -->\n",
       "<g id=\"edge12\" class=\"edge\">\n",
       "<title>array-012-&gt;array-021</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M503,-317.7C503,-309.98 503,-300.71 503,-292.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"506.5,-292.1 503,-282.1 499.5,-292.1 506.5,-292.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;028 -->\n",
       "<g id=\"node27\" class=\"node\">\n",
       "<title>array-028</title>\n",
       "<g id=\"a_node27\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (1, 1, 1)\n",
       "dtype: [('n', '&lt;i8'), ('total', '&lt;f8')]\n",
       "chunk memory: 16 bytes\n",
       "\n",
       "task memory: 48.0 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; mean -&gt; reduction -&gt; blockwise\n",
       "line: 7 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"462,-210 398,-210 398,-174 462,-174 462,-210\"/>\n",
       "<text text-anchor=\"middle\" x=\"430\" y=\"-195\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-028</text>\n",
       "<text text-anchor=\"middle\" x=\"430\" y=\"-184\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">mean (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;021&#45;&gt;array&#45;028 -->\n",
       "<g id=\"edge13\" class=\"edge\">\n",
       "<title>array-021-&gt;array-028</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M484.96,-245.7C475.94,-237.05 464.88,-226.45 455.06,-217.03\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"457.48,-214.5 447.84,-210.1 452.63,-219.55 457.48,-214.5\"/>\n",
       "</g>\n",
       "<!-- array&#45;022 -->\n",
       "<g id=\"node14\" class=\"node\">\n",
       "<title>array-022</title>\n",
       "<g id=\"a_node14\"><a xlink:title=\"shape: (999, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; empty -&gt; full\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"74,-426 0,-426 0,-390 74,-390 74,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"37\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-022</text>\n",
       "<text text-anchor=\"middle\" x=\"37\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;025 -->\n",
       "<g id=\"node21\" class=\"node\">\n",
       "<title>array-025</title>\n",
       "<g id=\"a_node21\"><a xlink:title=\"shape: (999, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "task memory: 32.0 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"234,-354 140,-354 140,-318 234,-318 234,-354\"/>\n",
       "<text text-anchor=\"middle\" x=\"187\" y=\"-339\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-025</text>\n",
       "<text text-anchor=\"middle\" x=\"187\" y=\"-328\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;022&#45;&gt;array&#45;025 -->\n",
       "<g id=\"edge14\" class=\"edge\">\n",
       "<title>array-022-&gt;array-025</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M73.69,-389.88C94.02,-380.39 119.47,-368.51 141.1,-358.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"142.84,-361.47 150.43,-354.07 139.88,-355.13 142.84,-361.47\"/>\n",
       "</g>\n",
       "<!-- array&#45;023 -->\n",
       "<g id=\"node15\" class=\"node\">\n",
       "<title>array-023</title>\n",
       "<g id=\"a_node15\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (10, 9, 8)\n",
       "dtype: int32\n",
       "chunk memory: 2.9 KB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; map_blocks -&gt; asarray\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"174,-498 100,-498 100,-462 174,-462 174,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"137\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-023</text>\n",
       "<text text-anchor=\"middle\" x=\"137\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;024 -->\n",
       "<g id=\"node16\" class=\"node\">\n",
       "<title>array-024</title>\n",
       "<g id=\"a_node16\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (1, 1, 1)\n",
       "dtype: int32\n",
       "chunk memory: 4 bytes\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; __getitem__ -&gt; index -&gt; map_direct -&gt; map_blocks -&gt; rechunk\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"181.5,-426 92.5,-426 92.5,-390 181.5,-390 181.5,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"137\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-024</text>\n",
       "<text text-anchor=\"middle\" x=\"137\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__getitem__ (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;023&#45;&gt;array&#45;024 -->\n",
       "<g id=\"edge15\" class=\"edge\">\n",
       "<title>array-023-&gt;array-024</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M137,-461.7C137,-453.98 137,-444.71 137,-436.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"140.5,-436.1 137,-426.1 133.5,-436.1 140.5,-436.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;024&#45;&gt;array&#45;025 -->\n",
       "<g id=\"edge16\" class=\"edge\">\n",
       "<title>array-024-&gt;array-025</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M149.36,-389.7C155.23,-381.47 162.37,-371.48 168.84,-362.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"171.82,-364.28 174.78,-354.1 166.12,-360.21 171.82,-364.28\"/>\n",
       "</g>\n",
       "<!-- array&#45;005 -->\n",
       "<g id=\"node17\" class=\"node\">\n",
       "<title>array-005</title>\n",
       "<g id=\"a_node17\"><a xlink:title=\"shape: (1000, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; empty -&gt; full\n",
       "line: 2 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"258.5,-498 199.5,-498 199.5,-462 258.5,-462 258.5,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"229\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-005</text>\n",
       "<text text-anchor=\"middle\" x=\"229\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;008 -->\n",
       "<g id=\"node20\" class=\"node\">\n",
       "<title>array-008</title>\n",
       "<g id=\"a_node20\"><a xlink:title=\"shape: (1000, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "task memory: 32.0 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 2 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"272.5,-426 199.5,-426 199.5,-390 272.5,-390 272.5,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"236\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-008</text>\n",
       "<text text-anchor=\"middle\" x=\"236\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;005&#45;&gt;array&#45;008 -->\n",
       "<g id=\"edge17\" class=\"edge\">\n",
       "<title>array-005-&gt;array-008</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M230.73,-461.7C231.5,-453.98 232.43,-444.71 233.29,-436.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"236.78,-436.4 234.29,-426.1 229.81,-435.71 236.78,-436.4\"/>\n",
       "</g>\n",
       "<!-- array&#45;006 -->\n",
       "<g id=\"node18\" class=\"node\">\n",
       "<title>array-006</title>\n",
       "<g id=\"a_node18\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (10, 9, 8)\n",
       "dtype: int32\n",
       "chunk memory: 2.9 KB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; asarray\n",
       "line: 2 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"340.5,-570 281.5,-570 281.5,-534 340.5,-534 340.5,-570\"/>\n",
       "<text text-anchor=\"middle\" x=\"311\" y=\"-555\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-006</text>\n",
       "<text text-anchor=\"middle\" x=\"311\" y=\"-544\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;007 -->\n",
       "<g id=\"node19\" class=\"node\">\n",
       "<title>array-007</title>\n",
       "<g id=\"a_node19\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (1, 1, 1)\n",
       "dtype: int32\n",
       "chunk memory: 4 bytes\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; rechunk\n",
       "line: 2 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"345,-498 277,-498 277,-462 345,-462 345,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"311\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-007</text>\n",
       "<text text-anchor=\"middle\" x=\"311\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;006&#45;&gt;array&#45;007 -->\n",
       "<g id=\"edge18\" class=\"edge\">\n",
       "<title>array-006-&gt;array-007</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M311,-533.7C311,-525.98 311,-516.71 311,-508.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"314.5,-508.1 311,-498.1 307.5,-508.1 314.5,-508.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;007&#45;&gt;array&#45;008 -->\n",
       "<g id=\"edge19\" class=\"edge\">\n",
       "<title>array-007-&gt;array-008</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M292.46,-461.7C283.2,-453.05 271.84,-442.45 261.74,-433.03\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"264.02,-430.37 254.33,-426.1 259.25,-435.49 264.02,-430.37\"/>\n",
       "</g>\n",
       "<!-- array&#45;008&#45;&gt;array&#45;025 -->\n",
       "<g id=\"edge20\" class=\"edge\">\n",
       "<title>array-008-&gt;array-025</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M223.89,-389.7C218.13,-381.47 211.14,-371.48 204.79,-362.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"207.58,-360.29 198.97,-354.1 201.84,-364.3 207.58,-360.29\"/>\n",
       "</g>\n",
       "<!-- array&#45;026 -->\n",
       "<g id=\"node26\" class=\"node\">\n",
       "<title>array-026</title>\n",
       "<g id=\"a_node26\"><a xlink:title=\"shape: (999, 900, 800)\n",
       "chunks: (100, 100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 8.0 MB\n",
       "\n",
       "task memory: 32.2 MB\n",
       "tasks: 720\n",
       "calls: &lt;module&gt; -&gt; __mul__ -&gt; elemwise -&gt; blockwise\n",
       "line: 6 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"421.5,-282 344.5,-282 344.5,-246 421.5,-246 421.5,-282\"/>\n",
       "<text text-anchor=\"middle\" x=\"383\" y=\"-267\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-026</text>\n",
       "<text text-anchor=\"middle\" x=\"383\" y=\"-256\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">__mul__ (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;025&#45;&gt;array&#45;026 -->\n",
       "<g id=\"edge21\" class=\"edge\">\n",
       "<title>array-025-&gt;array-026</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M234.45,-318.05C264.81,-307.21 304.09,-293.18 334.63,-282.28\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"335.97,-285.51 344.21,-278.85 333.62,-278.92 335.97,-285.51\"/>\n",
       "</g>\n",
       "<!-- array&#45;013 -->\n",
       "<g id=\"node22\" class=\"node\">\n",
       "<title>array-013</title>\n",
       "<g id=\"a_node22\"><a xlink:title=\"shape: (900, 800)\n",
       "chunks: (100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 80.0 KB\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; empty -&gt; full\n",
       "line: 4 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"349.5,-426 290.5,-426 290.5,-390 349.5,-390 349.5,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"320\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-013</text>\n",
       "<text text-anchor=\"middle\" x=\"320\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;016 -->\n",
       "<g id=\"node25\" class=\"node\">\n",
       "<title>array-016</title>\n",
       "<g id=\"a_node25\"><a xlink:title=\"shape: (900, 800)\n",
       "chunks: (100, 100)\n",
       "dtype: float64\n",
       "chunk memory: 80.0 KB\n",
       "\n",
       "task memory: 320.0 KB\n",
       "tasks: 72\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 4 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"419.5,-354 346.5,-354 346.5,-318 419.5,-318 419.5,-354\"/>\n",
       "<text text-anchor=\"middle\" x=\"383\" y=\"-339\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-016</text>\n",
       "<text text-anchor=\"middle\" x=\"383\" y=\"-328\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;013&#45;&gt;array&#45;016 -->\n",
       "<g id=\"edge22\" class=\"edge\">\n",
       "<title>array-013-&gt;array-016</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M335.57,-389.7C343.2,-381.22 352.52,-370.86 360.88,-361.58\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"363.52,-363.88 367.61,-354.1 358.31,-359.2 363.52,-363.88\"/>\n",
       "</g>\n",
       "<!-- array&#45;014 -->\n",
       "<g id=\"node23\" class=\"node\">\n",
       "<title>array-014</title>\n",
       "<g id=\"a_node23\"><a xlink:title=\"shape: (9, 8)\n",
       "chunks: (9, 8)\n",
       "dtype: int32\n",
       "chunk memory: 288 bytes\n",
       "\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; asarray\n",
       "line: 4 in &lt;module&gt;\">\n",
       "<polygon fill=\"none\" stroke=\"black\" points=\"431.5,-498 372.5,-498 372.5,-462 431.5,-462 431.5,-498\"/>\n",
       "<text text-anchor=\"middle\" x=\"402\" y=\"-483\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-014</text>\n",
       "<text text-anchor=\"middle\" x=\"402\" y=\"-472\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random </text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;015 -->\n",
       "<g id=\"node24\" class=\"node\">\n",
       "<title>array-015</title>\n",
       "<g id=\"a_node24\"><a xlink:title=\"shape: (9, 8)\n",
       "chunks: (1, 1)\n",
       "dtype: int32\n",
       "chunk memory: 4 bytes\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; random -&gt; map_direct -&gt; map_blocks -&gt; rechunk\n",
       "line: 4 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"436,-426 368,-426 368,-390 436,-390 436,-426\"/>\n",
       "<text text-anchor=\"middle\" x=\"402\" y=\"-411\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-015</text>\n",
       "<text text-anchor=\"middle\" x=\"402\" y=\"-400\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">random (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;014&#45;&gt;array&#45;015 -->\n",
       "<g id=\"edge23\" class=\"edge\">\n",
       "<title>array-014-&gt;array-015</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M402,-461.7C402,-453.98 402,-444.71 402,-436.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"405.5,-436.1 402,-426.1 398.5,-436.1 405.5,-436.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;015&#45;&gt;array&#45;016 -->\n",
       "<g id=\"edge24\" class=\"edge\">\n",
       "<title>array-015-&gt;array-016</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M397.3,-389.7C395.19,-381.9 392.64,-372.51 390.28,-363.83\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"393.64,-362.84 387.64,-354.1 386.88,-364.67 393.64,-362.84\"/>\n",
       "</g>\n",
       "<!-- array&#45;016&#45;&gt;array&#45;026 -->\n",
       "<g id=\"edge25\" class=\"edge\">\n",
       "<title>array-016-&gt;array-026</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M383,-317.7C383,-309.98 383,-300.71 383,-292.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"386.5,-292.1 383,-282.1 379.5,-292.1 386.5,-292.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;026&#45;&gt;array&#45;028 -->\n",
       "<g id=\"edge26\" class=\"edge\">\n",
       "<title>array-026-&gt;array-028</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M394.62,-245.7C400.14,-237.47 406.85,-227.48 412.93,-218.42\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"415.85,-220.36 418.52,-210.1 410.04,-216.46 415.85,-220.36\"/>\n",
       "</g>\n",
       "<!-- array&#45;029 -->\n",
       "<g id=\"node28\" class=\"node\">\n",
       "<title>array-029</title>\n",
       "<g id=\"a_node28\"><a xlink:title=\"shape: (10, 9, 8)\n",
       "chunks: (10, 9, 8)\n",
       "dtype: [('n', '&lt;i8'), ('total', '&lt;f8')]\n",
       "chunk memory: 11.5 KB\n",
       "\n",
       "task memory: 100.0 MB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; mean -&gt; reduction -&gt; rechunk\n",
       "line: 7 in &lt;module&gt;\">\n",
       "<polygon fill=\"#aaffc3\" stroke=\"black\" points=\"459.5,-138 400.5,-138 400.5,-102 459.5,-102 459.5,-138\"/>\n",
       "<text text-anchor=\"middle\" x=\"430\" y=\"-123\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-029</text>\n",
       "<text text-anchor=\"middle\" x=\"430\" y=\"-112\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">mean (rc)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;028&#45;&gt;array&#45;029 -->\n",
       "<g id=\"edge27\" class=\"edge\">\n",
       "<title>array-028-&gt;array-029</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M430,-173.7C430,-165.98 430,-156.71 430,-148.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"433.5,-148.1 430,-138.1 426.5,-148.1 433.5,-148.1\"/>\n",
       "</g>\n",
       "<!-- array&#45;032 -->\n",
       "<g id=\"node29\" class=\"node\">\n",
       "<title>array-032</title>\n",
       "<g id=\"a_node29\"><a xlink:title=\"shape: ()\n",
       "chunks: ()\n",
       "dtype: float64\n",
       "chunk memory: 8 bytes\n",
       "\n",
       "task memory: 23.1 KB\n",
       "tasks: 1\n",
       "calls: &lt;module&gt; -&gt; mean -&gt; reduction -&gt; squeeze -&gt; map_blocks -&gt; _map_blocks -&gt; blockwise\n",
       "line: 7 in &lt;module&gt;\">\n",
       "<polygon fill=\"#dcbeff\" stroke=\"black\" points=\"462,-66 398,-66 398,-30 462,-30 462,-66\"/>\n",
       "<text text-anchor=\"middle\" x=\"430\" y=\"-51\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">array-032</text>\n",
       "<text text-anchor=\"middle\" x=\"430\" y=\"-40\" font-family=\"Helvetica,sans-Serif\" font-size=\"10.00\">mean (bw)</text>\n",
       "</a>\n",
       "</g>\n",
       "</g>\n",
       "<!-- array&#45;029&#45;&gt;array&#45;032 -->\n",
       "<g id=\"edge28\" class=\"edge\">\n",
       "<title>array-029-&gt;array-032</title>\n",
       "<path fill=\"none\" stroke=\"black\" d=\"M430,-101.7C430,-93.98 430,-84.71 430,-76.11\"/>\n",
       "<polygon fill=\"black\" stroke=\"black\" points=\"433.5,-76.1 430,-66.1 426.5,-76.1 433.5,-76.1\"/>\n",
       "</g>\n",
       "</g>\n",
       "</svg>"
      ],
      "text/plain": [
       "<IPython.core.display.SVG object>"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "result.visualize()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "0b9b8905",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010904788970947266,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-019",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "9cbfd1d55dee4ccdb2899a3c7a82505e",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-019:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010373830795288086,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-003",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "41d6946ae6d54955a751c259ec5d92b3",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-003:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010881900787353516,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-011",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "30a02af2fb85414aae80cd58dda1ea86",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-011:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010125875473022461,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-024",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e597a5af3a3e41408dde687c242a7f0d",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-024:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010343313217163086,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-007",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d89a75e94e9a487ba54e05bb9121693e",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-007:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010203838348388672,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-015",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "11ccc063731f4e5f97ed72d09a335753",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-015:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010620832443237305,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-004",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "d165ac8f48934dc7b5088640e28daeb7",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-004:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010289907455444336,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-012",
       "rate": null,
       "total": 72,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "de5b105fd1cf4fe793dbd1fb7932687e",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-012:   0%|          | 0/72 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.009669065475463867,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-008",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a4010676e72141d4a9a43e17c7146fbb",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-008:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010063648223876953,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-016",
       "rate": null,
       "total": 72,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "884b264e385c4369b4d503b3442d8ad8",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-016:   0%|          | 0/72 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010417938232421875,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-020",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "4679db2472eb4f769172c296b96d2a0d",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-020:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010111331939697266,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-025",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "73eb74e1fdc149119ea5127a0a304c77",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-025:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.009608030319213867,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-021",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e2c058282fdd4cda94e0cd16ebde0f71",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-021:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.010048627853393555,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-026",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "59c2f6c51ff64630bb6272d4f1ce8847",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-026:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.00966501235961914,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-028",
       "rate": null,
       "total": 720,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "e103fce5d98a40e89896ba96bda6fd17",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-028:   0%|          | 0/720 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.009849071502685547,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-029",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "206f37e1325e4948b74ee63f6b830c95",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-029:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/json": {
       "ascii": false,
       "bar_format": null,
       "colour": null,
       "elapsed": 0.009930849075317383,
       "initial": 0,
       "n": 0,
       "ncols": null,
       "nrows": 48,
       "postfix": null,
       "prefix": "array-032",
       "rate": null,
       "total": 1,
       "unit": "it",
       "unit_divisor": 1000,
       "unit_scale": false
      },
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "a758526ad0c1460ab8cd5025595d73ab",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "array-032:   0%|          | 0/1 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/plain": [
       "array(0.49989647)"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "progress = TqdmProgressBar()\n",
    "result.compute(callbacks=[progress])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ce1b6bc5",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "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.8.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}