No test_model necessary in mpi_train method

Author: ASvyatkovskiy

plasma/models/mpi_runner.py

@@ -464,7 +464,7 @@ def mpi_make_predictions_and_evaluate(conf,shot_list,loader):
 def mpi_train(conf,shot_list_train,shot_list_validate,loader, callbacks_list=None):   
 
     specific_builder = builder.ModelBuilder(conf)
-    train_model,test_model = specific_builder.build_train_test_models()
+    train_model = specific_builder.build_model(False)
 
     #load the latest epoch we did. Returns -1 if none exist yet
     e = specific_builder.load_model_weights(train_model)