karma/crab
1
0
Fork 0
mirror of https://github.com/karma-riuk/crab.git synced 2025-07-04 13:18: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:19:36 +02:00
parent beaef7310f
commit c6650b2b87
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
extract_correct_predictions.py
View File

@ -16,8 +16,8 @@ class OutputType(Enum):
@dataclass @dataclass
class CommentGenSubmission: class CommentGenSubmission:
path: str path: str
line_from: int from_: int
line_to: Optional[int] to: Optional[int]
body: str body: str
@ -52,8 +52,8 @@ def extract_comment_predictions(dataset_path: str, output_path: str):
if sel and sel.comment_suggests_change: if sel and sel.comment_suggests_change:
results[entry.metadata.id] = CommentGenSubmission( results[entry.metadata.id] = CommentGenSubmission(
path=entry.comments[0].file, path=entry.comments[0].file,
line_from=entry.comments[0].from_, from_=entry.comments[0].from_,
line_to=entry.comments[0].to, to=entry.comments[0].to,
body=entry.comments[0].body, body=entry.comments[0].body,
) )