double_model.py

# double_model.py

class DoubleModel:
    def __init__(self):
        pass  # no training, it's just a rule

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