From 91440443b0ff0f4a10c829862200bb52ff198551 Mon Sep 17 00:00:00 2001
From: Jamesb <experimentallearning0@gmail.com>
Date: Tue, 7 Jan 2025 09:16:52 +0000
Subject: [PATCH] fix: types for the auto param in MIPRO

---
 dspy/teleprompt/mipro_optimizer_v2.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dspy/teleprompt/mipro_optimizer_v2.py b/dspy/teleprompt/mipro_optimizer_v2.py
index bd8eb48ad1..5ac7f1dfff 100644
--- a/dspy/teleprompt/mipro_optimizer_v2.py
+++ b/dspy/teleprompt/mipro_optimizer_v2.py
@@ -2,7 +2,7 @@
 import random
 import textwrap
 from collections import defaultdict
-from typing import Any, Callable, Dict, List, Optional, Tuple
+from typing import Any, Callable, Dict, List, Literal, Optional, Tuple
 
 import numpy as np
 import optuna
@@ -52,7 +52,7 @@ def __init__(
         teacher_settings: Dict = {},
         max_bootstrapped_demos: int = 4,
         max_labeled_demos: int = 16,
-        auto: Optional[str] = None,
+        auto: Optional[Literal["light", "medium", "heavy"]] = None,
         num_candidates: int = 10,
         num_threads: int = 6,
         max_errors: int = 10,