From ebd1d8680a4e8d10b1470fa450800ececeaa3afb Mon Sep 17 00:00:00 2001
From: cp3hnu <cp3hnu@gmail.com>
Date: Thu, 15 May 2025 15:22:36 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=85=A8=E9=80=89=E6=97=B6=E5=8F=AF?=
 =?UTF-8?q?=E4=BB=A5=E9=80=89=E4=B8=AD=E8=BF=90=E8=A1=8C=E7=9A=84=E5=AE=9E?=
 =?UTF-8?q?=E9=AA=8C=E5=AE=9E=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../components/CreateForm/ExecuteConfig.tsx      | 12 +++++++++++-
 .../ActiveLearn/components/CreateForm/utils.ts   |  4 ++++
 .../components/ExperimentInstanceList/index.tsx  | 16 ++++++++++++++--
 .../components/ExperimentInstanceList/index.tsx  | 16 ++++++++++++++--
 4 files changed, 43 insertions(+), 5 deletions(-)

diff --git a/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx b/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx
index c510dc3b..c5d1f476 100644
--- a/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx
+++ b/react-ui/src/pages/ActiveLearn/components/CreateForm/ExecuteConfig.tsx
@@ -17,6 +17,11 @@ import {
 
 function ExecuteConfig() {
   const form = Form.useFormInstance();
+  const task_type = Form.useWatch('task_type', form);
+  const queryStrategiesOptions =
+    task_type === AutoMLTaskType.Classification
+      ? queryStrategies.slice(0, 2)
+      : queryStrategies.slice(2);
   return (
     <>
       <SubAreaTitle
@@ -488,7 +493,12 @@ function ExecuteConfig() {
               },
             ]}
           >
-            <Select placeholder="请选择查询策略" options={queryStrategies} showSearch allowClear />
+            <Select
+              placeholder="请选择查询策略"
+              options={queryStrategiesOptions}
+              showSearch
+              allowClear
+            />
           </Form.Item>
         </Col>
       </Row>
diff --git a/react-ui/src/pages/ActiveLearn/components/CreateForm/utils.ts b/react-ui/src/pages/ActiveLearn/components/CreateForm/utils.ts
index 8bf7bce4..77bc517a 100644
--- a/react-ui/src/pages/ActiveLearn/components/CreateForm/utils.ts
+++ b/react-ui/src/pages/ActiveLearn/components/CreateForm/utils.ts
@@ -87,4 +87,8 @@ export const queryStrategies = [
     label: 'upper_confidence_bound',
     value: 'upper_confidence_bound',
   },
+  {
+    label: 'probability_of_improvement',
+    value: 'probability_of_improvement',
+  },
 ];
diff --git a/react-ui/src/pages/AutoML/components/ExperimentInstanceList/index.tsx b/react-ui/src/pages/AutoML/components/ExperimentInstanceList/index.tsx
index 05d6d786..d11afc9d 100644
--- a/react-ui/src/pages/AutoML/components/ExperimentInstanceList/index.tsx
+++ b/react-ui/src/pages/AutoML/components/ExperimentInstanceList/index.tsx
@@ -34,7 +34,14 @@ function ExperimentInstanceList({
 }: ExperimentInstanceListProps) {
   const { message } = App.useApp();
   const allIntanceIds = useMemo(() => {
-    return experimentInsList?.map((item) => item.id) || [];
+    return (
+      experimentInsList
+        ?.filter(
+          (item) =>
+            item.status !== ExperimentStatus.Running && item.status !== ExperimentStatus.Pending,
+        )
+        .map((item) => item.id) || []
+    );
   }, [experimentInsList]);
   const [
     selectedIns,
@@ -126,7 +133,12 @@ function ExperimentInstanceList({
     <div>
       <div className={styles.tableExpandBox} style={{ paddingBottom: '16px' }}>
         <div className={styles.check}>
-          <Checkbox checked={checked} indeterminate={indeterminate} onChange={checkAll}></Checkbox>
+          <Checkbox
+            checked={checked}
+            indeterminate={indeterminate}
+            disabled={allIntanceIds.length === 0}
+            onChange={checkAll}
+          ></Checkbox>
         </div>
         <div className={styles.index}>序号</div>
         <div className={styles.description}>运行时长</div>
diff --git a/react-ui/src/pages/Experiment/components/ExperimentInstanceList/index.tsx b/react-ui/src/pages/Experiment/components/ExperimentInstanceList/index.tsx
index 029a78c7..bcfbc3ff 100644
--- a/react-ui/src/pages/Experiment/components/ExperimentInstanceList/index.tsx
+++ b/react-ui/src/pages/Experiment/components/ExperimentInstanceList/index.tsx
@@ -39,7 +39,14 @@ function ExperimentInstanceList({
 }: ExperimentInstanceListProps) {
   const { message } = App.useApp();
   const allIntanceIds = useMemo(() => {
-    return experimentInsList?.map((item) => item.id) || [];
+    return (
+      experimentInsList
+        ?.filter(
+          (item) =>
+            item.status !== ExperimentStatus.Running && item.status !== ExperimentStatus.Pending,
+        )
+        .map((item) => item.id) || []
+    );
   }, [experimentInsList]);
   const [
     selectedIns,
@@ -127,7 +134,12 @@ function ExperimentInstanceList({
     <div>
       <div className={styles.tableExpandBox} style={{ paddingBottom: '16px' }}>
         <div className={styles.check}>
-          <Checkbox checked={checked} indeterminate={indeterminate} onChange={checkAll}></Checkbox>
+          <Checkbox
+            checked={checked}
+            indeterminate={indeterminate}
+            disabled={allIntanceIds.length === 0}
+            onChange={checkAll}
+          ></Checkbox>
         </div>
         <div className={styles.index}>序号</div>
         <div className={styles.tensorBoard}>可视化</div>