Update handler.py

Fixed error "setting an array element with a sequence. "

handler.py

@@ -76,10 +76,10 @@ class EndpointHandler:
             if k == 'instruction':
                 prompt += f'<{v}>' if isinstance(v, str) else "".join([f'<{v_i}>' for v_i in v])
             elif k == 'up':
-                if v and len(prompt_config['drug']) != 0:
+                if v:
                     prompt += f'{multi_modal_prefix}<{k}>{v} </{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
             elif k == 'down':
-                if v and "drug" in list(prompt_config.keys()):
+                if v:
                     prompt += f'{multi_modal_prefix}<{k}>{v} </{k}>' if isinstance(v, str) else f'{multi_modal_prefix}<{k}>{" ".join(v)} </{k}>'
             elif k == 'age':
                 if isinstance(v, int):
@@ -327,9 +327,13 @@ class EndpointHandler:
         acc_embs_up1 = []
         acc_embs_up2 = []
         if 'up' in config_data:
-            for gs in config_data['up']:
-                acc_embs_up1.append(self.emb_hgt_genes.get(gs))
-                acc_embs_up2.append(self.emb_gpt_genes.get(gs))
+            for gs in config_data['up']: 
+                try:
+                    acc_embs_up1.append(self.emb_hgt_genes[gs])
+                    acc_embs_up2.append(self.emb_gpt_genes[gs])
+                except Exception as e: 
+                    pass
+
         
         acc_embs_up1_mean = np.array(acc_embs_up1).mean(0) if acc_embs_up1 else None
         acc_embs_up2_mean = np.array(acc_embs_up2).mean(0) if acc_embs_up2 else None
@@ -338,8 +342,14 @@ class EndpointHandler:
         acc_embs_down2 = []
         if 'down' in config_data:
             for gs in config_data['down']:
-                acc_embs_down1.append(self.emb_hgt_genes.get(gs))
-                acc_embs_down2.append(self.emb_gpt_genes.get(gs))
+                try:
+                    acc_embs_down1.append(self.emb_hgt_genes[gs])
+                    acc_embs_down2.append(self.emb_gpt_genes[gs])
+                except Exception as e: 
+                    pass
+            # for gs in config_data['down']:
+            #     acc_embs_down1.append(self.emb_hgt_genes.get(gs))
+            #     acc_embs_down2.append(self.emb_gpt_genes.get(gs))
 
         acc_embs_down1_mean = np.array(acc_embs_down1).mean(0) if acc_embs_down1 else None
         acc_embs_down2_mean = np.array(acc_embs_down2).mean(0) if acc_embs_down2 else None