debug rapidapi.py

OpenBMB · Aug 8, 2023 · db505c5 · db505c5 · victor-lang · Aug 10, 2023
1 parent 654ad15
commit db505c5
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/toolbench/inference/Downstream_tasks/rapidapi.py b/toolbench/inference/Downstream_tasks/rapidapi.py
@@ -43,13 +43,12 @@ def get_white_list(tool_root_dir):
     return white_list
 
 def contain(candidate_list, white_list):
-    output = set()
+    output = []
     for cand in candidate_list:
-        if cand in white_list:
-            output.add(white_list[cand])
-        else:
+        if cand not in white_list.keys():
             return False
-    return list(output)
+        output.append(white_list[cand])
+    return output