class DoubleModel: def predict(self, number): return number * 2