seems to fix the broken tests
@@ -16,8 +16,8 @@ class SumLazyTensor(LazyTensor):
lazy_tensors = tuple(lazify(lt) for lt in lazy_tensors)
except TypeError:
raise TypeError("All arguments of a SumLazyTensor should be LazyTensors or Tensors")
- batch_shape = _mul_broadcast_shape(*[lt.shape for lt in lazy_tensors])
- lazy_tensors = tuple(lt.expand(batch_shape) for lt in lazy_tensors)
+ # batch_shape = _mul_broadcast_shape(*[lt.shape for lt in lazy_tensors])
+ # lazy_tensors = tuple(lt.expand(batch_shape) for lt in lazy_tensors)
super(SumLazyTensor, self).__init__(*lazy_tensors, **kwargs)
self.lazy_tensors = lazy_tensors
