Replace Diagnostic with Callback in Cache

dask · Aug 7, 2015 · 2ff2f63 · 2ff2f63
1 parent 3584f1f
commit 2ff2f63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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