Skip to content

Commit 947296b

Browse files
authored
Fix type annotation errors in llm_generator.py (#74)
Added explicit list[DatasetCase] type annotations to the cases variable in both LlamaDatasetGenerator.generate() and OpenAIDatasetGenerator.generate() methods to satisfy Mypy type checking. Closes #67
1 parent 8fce886 commit 947296b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/agentunit/generators/llm_generator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ async def generate(
141141

142142
cases_data = json.loads(response_text)
143143

144-
cases = []
144+
cases: list[DatasetCase] = []
145145
for case_data in cases_data:
146146
case = DatasetCase(
147147
id=case_data.get("id", f"generated_{len(cases)}"),
@@ -288,7 +288,7 @@ async def generate(
288288

289289
cases_data = json.loads(response_text)
290290

291-
cases = []
291+
cases: list[DatasetCase] = []
292292
for case_data in cases_data:
293293
case = DatasetCase(
294294
id=case_data.get("id", f"generated_{len(cases)}"),

0 commit comments

Comments
 (0)