double_model.py

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