Small bug fixed in name_test_dataset (train fct)

Signed-off-by: ElinaFF <[email protected]>
plpla · Jul 4, 2018 · 41ae454 · 41ae454
1 parent 4ed6de8
commit 41ae454
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/interface/command_line_tool.py b/interface/command_line_tool.py
@@ -229,15 +229,13 @@ def train(self):
 
         if args.pnnl:
             dt_list.append("PNL")
-            name_test_dataset.append("PNL")
-
+
         if args.mclean:
             dt_list.append("McLean")
-            name_test_dataset.append("McLean")
-
+
         if args.cbm:
             dt_list.append("CBM")
-            name_test_dataset.append("CBM")
+
         logging.debug("Number of training dataset: {}".format(len(training_datasets)))
 
         logging.debug("Training dataset list: {}".format(dt_list))