fix scene base 'of_mode' error (#881)

Co-authored-by: jcfwwangxun_qm <jcfwwangxun_qm@faw.com.cn>
eosphoros-ai · Dec 2, 2023 · 13fb9d0 · 13fb9d0
1 parent a35b612
commit 13fb9d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pilot/scene/base.py b/pilot/scene/base.py
@@ -109,7 +109,7 @@ class ChatScene(Enum):
 
     @staticmethod
     def of_mode(mode):
-        return [x for x in ChatScene._value_ if x.code == mode][0]
+        return [x for x in ChatScene if mode == x.value()][0]
 
     @staticmethod
     def is_valid_mode(mode):