Merge pull request #660 from guardian/tf-bandit-data-refactor

support more than 1 bandit methodology
guardian · Dec 12, 2024 · d6d83a3 · d6d83a3
2 parents 0583108 + 672c580
commit d6d83a3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/public/src/components/channelManagement/TestMethodologyEditor.tsx b/public/src/components/channelManagement/TestMethodologyEditor.tsx
@@ -16,6 +16,8 @@ import AddIcon from '@mui/icons-material/Add';
 import Alert from '@mui/lab/Alert';
 import { addMethodologyToTestName } from './helpers/methodology';
 
+const isBandit = (methodology: Methodology): boolean => methodology.name === 'EpsilonGreedyBandit';
+
 const useStyles = makeStyles(({ spacing, palette }: Theme) => ({
   container: {
     '& > * + *': {
@@ -122,9 +124,9 @@ const TestMethodology: React.FC<TestMethodologyProps> = ({
               }}
             />
           </div>
-          <BanditAnalyticsButton testName={testName} channel={channel} />
         </>
       )}
+      {isBandit(methodology) && <BanditAnalyticsButton testName={testName} channel={channel} />}
       <div className={classes.testNameAndDeleteButton}>
         {methodology.testName && <div className={classes.testName}>{methodology.testName}</div>}
         <div className={classes.deleteButton}>