def model_fn()

in src/inference.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def model_fn(model_dir):
    """
    Deserialize and return fitted model.
    """
    model_file = "xgboost-model"
    booster = pkl.load(open(os.path.join(model_dir, model_file), "rb"))
    return booster