diff --git a/modules/deeplearning/unet_l1b_l2.py b/modules/deeplearning/unet_l1b_l2.py
index d3084cf334e0ea32a5dda4741de64a58ce538bd7..e4a025a6c46df0d6bdb20e2baf34ea8352d9d57f 100644
--- a/modules/deeplearning/unet_l1b_l2.py
+++ b/modules/deeplearning/unet_l1b_l2.py
@@ -887,14 +887,7 @@ class UNET:
             preds = np.argmax(preds, axis=1)
         self.test_preds = preds
 
-    def run(self, filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst):
-        self.setup_pipeline(filename_l1b_trn, filename_l1b_tst, filename_l2_trn, filename_l2_tst)
-        self.build_model()
-        self.build_training()
-        self.build_evaluation()
-        self.do_training()
-
-    def run_test(self, directory):
+    def run(self, directory):
         train_data_files = glob.glob(directory+'data_train*.npy')
         valid_data_files = glob.glob(directory+'data_valid*.npy')
         train_label_files = glob.glob(directory+'label_train*.npy')