def output_type(self):
    if isinstance(self.arg, ir.BooleanValue):
        dtype = dt.int64
    else:
        dtype = self.arg.type().largest
    return dtype.column_type()
def output_type(self):
    if isinstance(self.arg, ir.BooleanValue):
        dtype = dt.int64
    else:
        dtype = self.arg.type().largest
    return dtype.column_type()