karma/crab-webapp
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab-webapp.git synced 2025-07-05 06:08:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

made code more consistent

Browse Source
This commit is contained in:
Karma Riuk
2025-06-23 11:23:28 +02:00
parent 18c1c32a8c
commit 9ddb0fcb7e
3 changed files with 23 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/utils/dataset.py
View File

@ -51,8 +51,8 @@ class Metadata:
@dataclass
class CommentGenSubmission:
path: str
line_from: int
line_to: Optional[int]
from_: Optional[int]
to: int
body: str
@classmethod
@ -61,15 +61,13 @@ class CommentGenSubmission:
raise InvalidJsonFormatError("Submitted json doesn't contain an object")
if not all(k in data and isinstance(data[k], str) for k in ["path", "body"]):
raise InvalidJsonFormatError("Submitted json doesn't contain the required fields")
if not all(
k in data and isinstance(data[k], (int, type(None))) for k in ["line_from", "line_to"]
):
if not all(k in data and isinstance(data[k], (int, type(None))) for k in ["from_", "to"]):
raise InvalidJsonFormatError("Submitted json doesn't contain the required fields")
return cls(
path=data["path"],
line_from=data["line_from"],
line_to=data.get("line_to"),
from_=data.get("from_"),
to=data["to"],
body=data["body"],
)