customer_model.py

import ...
import fire


class CustomerModel(BaseModel):

    def __init__(self):
        super().__init__()

    def train_model(lookback_days=30: int):
        """
        Trains a model using data from the past <lookback_days> and persists to a model store
        """
        ...

    def score_customers(model_version=None: str, lookback_days=30: int):
        """
        Pulls <model_version> from the model store and uses it to score production data for the last <lookback_days>
        """
        ...

if __name__ == "main":
    fire.Fire(CustomerModel)