Remove cull from SubgraphCallable

mrocklin · Nov 26, 2018 · d35fe44 · d35fe44
1 parent 722b87b
commit d35fe44
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/dask/optimization.py b/dask/optimization.py
@@ -916,7 +916,7 @@ class SubgraphCallable(object):
     __slots__ = ('dsk', 'outkey', 'inkeys', 'name')
 
     def __init__(self, dsk, outkey, inkeys, name='subgraph_callable'):
-        self.dsk, _ = cull(dsk, outkey)
+        self.dsk = dsk
         self.outkey = outkey
         self.inkeys = inkeys
         self.name = name

diff --git a/dask/tests/test_optimization.py b/dask/tests/test_optimization.py
@@ -1089,10 +1089,6 @@ def f(*args):
     })
 
 
-def dontcall(x):
-    raise ValueError("shouldn't be called")
-
-
 def func_with_kwargs(a, b, c=2):
     return a + b + c
 
@@ -1108,7 +1104,6 @@ def test_SubgraphCallable():
            'd': (inc, 'a'),
            'e': (add, 'c', 'd'),
            'f': ['a', 2, 'b', (add, 'b', (sum, non_hashable))],
-           'g': (dontcall, 'in1'),
            'h': (add, (sum, 'f'), (sum, ['a', 'b']))}
 
     f = SubgraphCallable(dsk, 'h', ['in1', 'in2'], name='test')