Can we change this to add a bit more information?

`raise TypeError("Expected ndarray or DistArray, got: %s" % type(data))`