bugfix: printing only for rank 0

ACEsuit · Jun 12, 2024 · cf56258 · cf56258
1 parent 7c043e1
commit cf56258
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mace/cli/preprocess_data_mpi.py b/mace/cli/preprocess_data_mpi.py
@@ -312,8 +312,9 @@ def main() -> None:
 
     if args.valid_file is not None:
         valid_configs = read_configs(args.valid_file, args, config_type_weights)
-        logging.info(f"Number of training configurations: {len(train_configs)}")
-        logging.info(f"Number of validation configurations: {len(valid_configs)}")
+        if rank == 0:
+            logging.info(f"Number of training configurations: {len(train_configs)}")
+            logging.info(f"Number of validation configurations: {len(valid_configs)}")
     else:
         if rank == 0:
             logging.info(