fix schemas

This commit is contained in:
jxxghp
2023-07-07 17:48:47 +08:00
parent 9ccb2b482e
commit de93b18673
3 changed files with 19 additions and 7 deletions

View File

@ -1,4 +1,4 @@
from dataclasses import dataclass
from dataclasses import dataclass, asdict
from typing import Union, Optional, List
import cn2an
@ -439,3 +439,11 @@ class MetaBase(object):
elif str(ep).isdigit():
self.begin_episode = int(ep)
self.end_episode = None
def to_dict(self):
"""
转为字典
"""
dicts = asdict(self)
dicts["type"] = self.type.value if self.type else None
return dicts