def output_type(self):
    if isinstance(self.arg.type(), dt.Decimal):
        return self.arg._factory
    return rlz.shape_like(self.arg, dt.int64)
def output_type(self):
    if isinstance(self.arg.type(), dt.Decimal):
        return self.arg._factory
    return rlz.shape_like(self.arg, dt.int64)