From 2ff2f63a963b1e240443576d8ef733b1f719f9fe Mon Sep 17 00:00:00 2001
From: Matthew Rocklin <mrocklin@gmail.com>
Date: Fri, 7 Aug 2015 11:47:02 -0700
Subject: [PATCH] Replace Diagnostic with Callback in Cache

---
 dask/diagnostics/cache.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dask/diagnostics/cache.py b/dask/diagnostics/cache.py
index e2c65a6fc3fc..9a38ea8bd57e 100644
--- a/dask/diagnostics/cache.py
+++ b/dask/diagnostics/cache.py
@@ -1,4 +1,4 @@
-from .core import Diagnostic
+from ..callbacks import Callback
 from timeit import default_timer
 from numbers import Number
 
@@ -7,7 +7,7 @@
 except ImportError:
     pass
 
-class Cache(Diagnostic):
+class Cache(Callback):
     """ Use cache for computation
 
     Example