def square(x: DataPoint) -> DataPoint: fields = x.asDict() fields["num_squared"] = x.num**2 return Row(**fields)
def square(x: DataPoint) -> DataPoint: fields = x.asDict() fields["num_squared"] = square_hit_tracker(x.num) return Row(**fields)