berka.py

 from mostlyai import MostlyAI

# initialize client
mostly = MostlyAI()

# train generator for BERKA dataset
config = {
    "name": "berka",
    "tables": [
        {
            "name": "account", 
            "data": "https://github.com/mostly-ai/public-demo-data/raw/dev/berka/data/account.csv.gz",
            "primaryKey": "account_id"
        },
        {
            "name": "transaction",
            "data": "https://github.com/mostly-ai/public-demo-data/raw/dev/berka/data/trans.csv.gz",
            "primaryKey": "trans_id",
            "foreignKeys": [{"column": "account_id", "referencedTable": "account", "isContext": True}]
        }
    ]
}
g = mostly.train(config=config)