We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents d5fe6aa + 1b2793b commit bcd64fcCopy full SHA for bcd64fc
src/amzn_nova_prompt_optimizer/core/evaluation/__init__.py
@@ -65,7 +65,13 @@ def _get_or_run_inference(self, model_id: str) -> list:
65
66
if cache_key not in self._inference_cache:
67
self.logger.info("Cache miss - Running new inference on Dataset")
68
- self._inference_cache[cache_key] = self.inference_runner.run(model_id)
+ 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
+
75
else:
76
self.logger.info("Using cached inference results")
77
0 commit comments