Simplify B.cast for TF

wesselb · Jun 4, 2018 · 073d6dd · 073d6dd
1 parent 48ddf32
commit 073d6dd
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/lab/proxies/tensorflow.py b/lab/proxies/tensorflow.py
@@ -26,17 +26,11 @@ def dtype(a):
     return a.dtype
 
 
-@_dispatch(_Numeric, object)
+@_dispatch(_Numeric, [object])
 def cast(a, dtype):
     return tf.cast(a, dtype)
 
 
-@_dispatch(object)
-def cast(a, dtype):
-    # This is a call with a keyword argument.
-    return cast(a, dtype)
-
-
 @_dispatch(object)
 def shape(a):
     return np.shape(a)