What is the correct gradient at zero for `pointwise.unit_step`?

I just noticed that the gradient of the pointwise function unit_step is set to be always zero.

Is that correct, or should we return np.nan if the input is exactly zero, like for examples sign does?