#!/usr/bin/env python3
"""Run a model."""

import os
import argparse
from icenet import MODELS
    

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument(
            '-d',
            required=True,
            help="Directory containing {train,test}.json."
    )
    parser.add_argument(
            'target_fn',
            help=("Target <module>.<fn> in the icenet package. "
                  "The function must take the data directory as argument.")
    )
    args = parser.parse_args()
    if not os.path.isdir(args.d):
        raise Exception("Output directory does not exist")

    model, fn = args.target_fn.split('.')
    print("Running model %s ..." % model)
    fn = getattr(MODELS[model], fn)
    fn(args.d)
    print("Done.")