From 69db08779a8054709aa09753cfc8cd29214cca8f Mon Sep 17 00:00:00 2001
From: liuht <liuht@lamda.nju.edu.cn>
Date: Sun, 14 Jan 2024 19:04:04 +0800
Subject: [PATCH] [FIX] move skip_test into workflow.py

---
 examples/dataset_table_workflow/hetero.py   |  2 +-
 examples/dataset_table_workflow/homo.py     |  2 +-
 examples/dataset_table_workflow/workflow.py | 12 ++++++------
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/examples/dataset_table_workflow/hetero.py b/examples/dataset_table_workflow/hetero.py
index 4aa206e..0099e63 100644
--- a/examples/dataset_table_workflow/hetero.py
+++ b/examples/dataset_table_workflow/hetero.py
@@ -106,7 +106,7 @@ class HeterogeneousDatasetWorkflow(TableWorkflow):
         )
 
 
-    def labeled_hetero_table_example(self, skip_test=False):
+    def labeled_hetero_table_example(self, skip_test):
         logger.info("Total Items: %d" % len(self.market))
         methods = ["user_model", "hetero_single_aug", "hetero_multiple_avg", "hetero_ensemble_pruning"]
         recorders = {method: Recorder() for method in methods}
diff --git a/examples/dataset_table_workflow/homo.py b/examples/dataset_table_workflow/homo.py
index 0b44f8b..b22f4e0 100644
--- a/examples/dataset_table_workflow/homo.py
+++ b/examples/dataset_table_workflow/homo.py
@@ -105,7 +105,7 @@ class HomogeneousDatasetWorkflow(TableWorkflow):
         )
         
         
-    def labeled_homo_table_example(self, skip_test=False):
+    def labeled_homo_table_example(self, skip_test):
         logger.info("Total Item: %d" % (len(self.market)))
         methods = ["user_model", "homo_single_aug", "homo_ensemble_pruning"]
         recorders = {method: Recorder() for method in methods}
diff --git a/examples/dataset_table_workflow/workflow.py b/examples/dataset_table_workflow/workflow.py
index e079b6e..93a6fcc 100644
--- a/examples/dataset_table_workflow/workflow.py
+++ b/examples/dataset_table_workflow/workflow.py
@@ -18,25 +18,25 @@ class TableDatasetWorkflow:
         )
         workflow.unlabeled_homo_table_example()
 
-    def labeled_homo_table_example(self):
+    def labeled_homo_table_example(self, skip_test=False):
         workflow = HomogeneousDatasetWorkflow(
             benchmark_config=homo_table_benchmark_config,
             name="easy",
             rebuild=False
         )
-        workflow.labeled_homo_table_example()
+        workflow.labeled_homo_table_example(skip_test=skip_test)
     
     def cross_feat_eng_hetero_table_example(self):
         set_seed(0)
         workflow = HeterogeneousDatasetWorkflow(
             benchmark_config=hetero_cross_feat_eng_benchmark_config,
             name="hetero",
-            rebuild=False,
-            retrain=False
+            rebuild=True,
+            retrain=True
         )
         workflow.unlabeled_hetero_table_example()
 
-    def cross_task_hetero_table_example(self):
+    def cross_task_hetero_table_example(self, skip_test=False):
         set_seed(0)
         workflow = HeterogeneousDatasetWorkflow(
             benchmark_config=hetero_cross_task_benchmark_config,
@@ -44,7 +44,7 @@ class TableDatasetWorkflow:
             rebuild=False,
             retrain=False
         )
-        workflow.labeled_hetero_table_example()
+        workflow.labeled_hetero_table_example(skip_test=skip_test)
 
 
 if __name__ == "__main__":