From 591d6d552e00396b0b8f8101d52e6ee31a7f4128 Mon Sep 17 00:00:00 2001
From: jon-lee <jon-lee@sjtu.edu.cn>
Date: Tue, 25 Feb 2025 14:15:58 +0800
Subject: [PATCH] fix: add model_config in Stage class
---
joj3_config_generator/models/task.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/joj3_config_generator/models/task.py b/joj3_config_generator/models/task.py
index e5ffdaa..87c5b40 100644
--- a/joj3_config_generator/models/task.py
+++ b/joj3_config_generator/models/task.py
@@ -82,8 +82,7 @@ class Stage(BaseModel):
cases: Optional[Dict[str, "Stage"]] = None
diff: Optional[ParserDiff] = ParserDiff()
- class Config:
- extra = "allow"
+ model_config = {"extra": "allow"}
@model_validator(mode="before")
@classmethod