minor changes

alibaba · Mar 30, 2023 · 86f4938 · 86f4938
1 parent 30304d5
commit 86f4938
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_tree_based_model_for_vfl.py b/tests/test_tree_based_model_for_vfl.py
@@ -157,7 +157,7 @@ def set_config_for_rf_base(self, cfg):
 
         return backup_cfg
 
-    def set_config_for_rf_label_base(self, cfg):
+    def set_config_for_rf_label_scattering(self, cfg):
         backup_cfg = cfg.clone()
 
         import torch
@@ -473,7 +473,7 @@ def test_RF_Base(self):
 
     def test_RF_lable_Base(self):
         init_cfg = global_cfg.clone()
-        backup_cfg = self.set_config_for_rf_label_base(init_cfg)
+        backup_cfg = self.set_config_for_rf_label_scattering(init_cfg)
         setup_seed(init_cfg.seed)
         update_logger(init_cfg, True)