add option pt_train_file == "mp"

Author: ilyes319

mace/cli/run_train.py

@@ -265,7 +265,7 @@ def run(args: argparse.Namespace) -> None:
         args.loss = "universal"
         if (
             args.foundation_model in ["small", "medium", "large"]
-            or args.pt_train_file is None
+            or args.pt_train_file == "mp"
         ):
             logging.info(
                 "Using foundation model for multiheads finetuning with Materials Project data"