def square_returns_none(x: DataPoint) -> DataPoint: if x.num == 2: return None x.num = x.num**2 return x
def square(x: DataPoint) -> DataPoint: x.num = x.num**2 return x