From a27ad6b49fd04702526926bd142ddd210ac16846 Mon Sep 17 00:00:00 2001
From: Barry57 <160387047+Barry57@users.noreply.github.com>
Date: Mon, 28 Oct 2024 00:26:55 +0800
Subject: [PATCH] Update test_scalar_ge.py

---
 pytest/test_scalar_ge.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pytest/test_scalar_ge.py b/pytest/test_scalar_ge.py
index a86aa93..269ef6d 100644
--- a/pytest/test_scalar_ge.py
+++ b/pytest/test_scalar_ge.py
@@ -16,12 +16,12 @@ def test_func_ge():
                               num_hidden_layers=2, nodes_hidden_layer=[1000,100], 
                               Learning_Rate2=0.035, L2=0.01, Learning_Rate1=0.06, L=0.09, 
                               Num_Epochs=50, t = 0.01, model = None, split_type = 1, ratio = [7, 2, 1], 
-                              important_feature = False, plot = False, model_reg = None, isfunc = False)
+                              important_feature = True, plot = False, model_reg = None, isfunc = False)
   scalar_ge_res_3 = scalar_ge(data=scalar_survival[0], ytype='Survival', dim_G=500, dim_E=5, haveGE=False, 
                               num_hidden_layers=2, nodes_hidden_layer=[1000,100], 
                               Learning_Rate2=0.035, L2=0.01, Learning_Rate1=0.06, L=0.09, 
                               Num_Epochs=50, t = 0.01, model = None, split_type = 1, ratio = [7, 2, 1], 
-                              important_feature = False, plot = False, model_reg = None, isfunc = False)
+                              important_feature = True, plot = False, model_reg = None, isfunc = False)
   assert scalar_ge_res_1 is not None
   assert scalar_ge_res_2 is not None
   assert scalar_ge_res_3 is not None