def mpnn_model_builder(model_params , model_dir): return MPNNModel(**model_params, model_dir = "./models")
def mpnn_model_builder(model_params, model_dir): mp_model = MPNNModel(**model_params, model_dir="./test_models") return mp_model