Merge pull request #76 from heyalexchoi/fix-propose-key-error

Skip components that are not present in reflective dataset instead of raising KeyError failure

Author: LakshyAAAgrawal

src/gepa/proposer/reflective_mutation/reflective_mutation.py

@@ -70,6 +70,13 @@ def propose_new_texts(
             raise ValueError("reflection_lm must be provided when adapter.propose_new_texts is None.")
         new_texts: dict[str, str] = {}
         for name in components_to_update:
+            # Gracefully handle cases where a selected component has no data in reflective_dataset
+            if name not in reflective_dataset or not reflective_dataset.get(name):
+                self.logger.log(
+                    f"Component '{name}' is not in reflective dataset. Skipping."
+                )
+                continue
+
             base_instruction = candidate[name]
             dataset_with_feedback = reflective_dataset[name]
             new_texts[name] = InstructionProposalSignature.run(