fixed adding cache argument to certain algorithms (#471)

refuel-ai · Jul 20, 2023 · a7b4ce8 · a7b4ce8
1 parent 740122c
commit a7b4ce8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/autolabel/few_shot/__init__.py b/src/autolabel/few_shot/__init__.py
@@ -104,6 +104,10 @@ def initialize_selector(
 
         example_cls = ALGORITHM_TO_IMPLEMENTATION[algorithm]
 
-        params["cache"] = cache
+        if algorithm not in [
+            FewShotAlgorithm.FIXED,
+            FewShotAlgorithm.LABEL_DIVERSITY_RANDOM,
+        ]:
+            params["cache"] = cache
 
         return example_cls.from_examples(**params)