Skip to content

Commit bcd64fc

Browse files
authored
Merge pull request #42 from malte-aws/fix-empty-inference-cache
fix: empty inference cache
2 parents d5fe6aa + 1b2793b commit bcd64fc

File tree

1 file changed

+7
-1
lines changed
  • src/amzn_nova_prompt_optimizer/core/evaluation

1 file changed

+7
-1
lines changed

src/amzn_nova_prompt_optimizer/core/evaluation/__init__.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,13 @@ def _get_or_run_inference(self, model_id: str) -> list:
6565

6666
if cache_key not in self._inference_cache:
6767
self.logger.info("Cache miss - Running new inference on Dataset")
68-
self._inference_cache[cache_key] = self.inference_runner.run(model_id)
68+
inference_results = self.inference_runner.run(model_id)
69+
if inference_results and len(inference_results) > 0:
70+
self._inference_cache[cache_key] = inference_results
71+
else:
72+
self.logger.warning("No inference results returned. Check the inference logs for any errors.")
73+
return inference_results
74+
6975
else:
7076
self.logger.info("Using cached inference results")
7177

0 commit comments

Comments
 (0)