fix unnecessary mlflow dep (#7942)

Author: TomeHirata

dspy/teleprompt/mipro_optimizer_v2.py

@@ -649,8 +649,6 @@ def _log_minibatch_eval(
         trajectory = "[" + full_eval_scores + "]"
         logger.info(f"Full eval scores so far: {trajectory}")
         logger.info(f"Best full score so far: {best_score}")
-        import mlflow
-        mlflow.log_metric("best score", best_score, step=trial.number)
         logger.info(
             f"{'=' * len(f'== Trial {trial.number + 1} / {adjusted_num_trials} - Minibatch Evaluation ==')}\n\n"
         )
@@ -681,8 +679,6 @@ def _log_normal_eval(
         full_eval_scores = ", ".join([f"{s['score']}" for s in score_data if s["full_eval"]])
         logger.info(f"Scores so far: {'[' + full_eval_scores + ']'}")
         logger.info(f"Best score so far: {best_score}")
-        import mlflow
-        mlflow.log_metric("best score", best_score, step=trial.number)
         logger.info(f"{'=' * len(f'===== Trial {trial.number + 1} / {num_trials} =====')}\n\n")
 
     def _select_and_insert_instructions_and_demos(

dspy/utils/callback.py

@@ -241,7 +241,7 @@ def on_evaluate_start(
     def on_evaluate_end(
         self,
         call_id: str,
-        outputs: Optional[Dict[str, Any]],
+        outputs: Optional[Any],
         exception: Optional[Exception] = None,
     ):
         """A handler triggered after evaluation is executed.