@@ -3,6 +3,10 @@
 """
 
 from dask import delayed
+import logging
+
+log = logging.getLogger(__name__)
+
 
 class ARLExecuteBase():
 
@@ -23,7 +27,7 @@ def execute(self, func, *args, **kwargs):
             return func
 
     def type(self):
-        """ Get the name of the execution system
+        """ Get the type of the execution system
         
         :return:
         """
@@ -39,7 +43,13 @@ def get(self, value):
         :return:
         """
         if self.use_dask:
-            return value.compute()
+            log.debug("arlexecute.get: Executing %d nodes in graph" % len(value.dask.dicts))
+            import time
+            start=time.time()
+            result = value.compute()
+            duration = time.time()-start
+            log.debug("arlexecute.get: Execution took %.3f seconds" % duration)
+            return result
         else:
             return value
 
 
@@ -21,21 +21,14 @@
     "\n",
     "import numpy\n",
     "\n",
-    "sys.path.append(os.path.join('..', '..'))\n",
-    "\n",
     "import astropy.units as u\n",
-    "\n",
-    "from arl.graphs.execute import arlexecute\n",
-    "\n",
     "from astropy.coordinates import SkyCoord\n",
     "\n",
+    "from arl.graphs.execute import arlexecute\n",
     "from arl.data.polarisation import PolarisationFrame\n",
     "from arl.graphs.generic_graphs import create_generic_image_graph\n",
     "from arl.graphs.dask_init import get_dask_Client\n",
-    "from arl.image.iterators import image_raster_iter\n",
-    "from arl.image.operations import qa_image\n",
-    "from arl.util.testing_support import create_test_image\n",
-    "\n"
+    "from arl.util.testing_support import create_test_image"
    ]
   },
   {
@@ -48,14 +41,10 @@
    "outputs": [],
    "source": [
     "frequency = numpy.array([1e8])\n",
-    "\n",
     "phasecentre = SkyCoord(ra=+15.0 * u.deg, dec=-35.0 * u.deg, frame='icrs', equinox='J2000')\n",
-    "test_image = create_test_image(frequency=frequency, phasecentre=phasecentre,\n",
-    "                               cellsize=0.001,\n",
-    "                               polarisation_frame=PolarisationFrame('stokesI'))\n",
-    "\n",
-    "image_graph = arlexecute.execute(create_test_image)(frequency=frequency, phasecentre=phasecentre, cellsize=0.001,\n",
-    "                                         polarisation_frame=PolarisationFrame('stokesI'))"
+    "model_graph = arlexecute.execute(create_test_image)(frequency=frequency, phasecentre=phasecentre, cellsize=0.001,\n",
+    "                                         polarisation_frame=PolarisationFrame('stokesI'))\n",
+    "model = arlexecute.get(model_graph)\n"
    ]
   },
   {
@@ -77,11 +66,7 @@
     "    im.data = numpy.sqrt(numpy.abs(im.data))\n",
     "    return im\n",
     "\n",
-    "root = create_generic_image_graph(imagerooter, test_image, facets=4)\n",
-    "if arlexecute.use_dask:\n",
-    "    root.visualize()\n",
-    "\n",
-    "print(root)"
+    "root = create_generic_image_graph(imagerooter, model_graph, facets=4)"
    ]
   },
   {
@@ -98,7 +83,7 @@
    "outputs": [],
    "source": [
     "result = arlexecute.get(root)\n",
-    "numpy.testing.assert_array_almost_equal_nulp(result.data**2, numpy.abs(test_image.data), 7)"
+    "numpy.testing.assert_array_almost_equal_nulp(result.data**2, numpy.abs(model.data), 7)"
    ]
   },
   {
@@ -116,9 +101,16 @@
    "source": [
     "client = get_dask_Client()\n",
     "result=client.compute(root, sync=True)\n",
-    "numpy.testing.assert_array_almost_equal_nulp(result.data**2, numpy.abs(test_image.data), 7)\n",
+    "numpy.testing.assert_array_almost_equal_nulp(result.data**2, numpy.abs(model.data), 7)\n",
     "client.close()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {