diff --git a/dask_cuda/proxy_object.py b/dask_cuda/proxy_object.py index cb61ca1d4..0a69e5cad 100644 --- a/dask_cuda/proxy_object.py +++ b/dask_cuda/proxy_object.py @@ -441,14 +441,12 @@ def __array_ufunc__(self, ufunc, method, *inputs, **kwargs): o._obj_pxy_deserialize() if isinstance(o, ProxyObject) else o for o in inputs ) - kwargs = dict( - { - key: value._obj_pxy_deserialize() - if isinstance(value, ProxyObject) - else value - for key, value in kwargs.items() - } - ) + kwargs = { + key: value._obj_pxy_deserialize() + if isinstance(value, ProxyObject) + else value + for key, value in kwargs.items() + } return self._obj_pxy_deserialize().__array_ufunc__( ufunc, method, *inputs, **kwargs )