changed the structure of the dataset

This commit is contained in:
Karma Riuk
2025-03-17 15:26:45 +01:00
parent b2d7377d1f
commit 3d9b386f22

View File

@ -1,5 +1,5 @@
from dataclasses import dataclass, field
from typing import List
from typing import Dict, List
import json
@dataclass
@ -25,10 +25,10 @@ class Diff:
@dataclass
class DatasetEntry:
metadata: Metadata
files: List[FileData] # files before the PR (before the first PR commits)
diffs_before: List[Diff] # diffs between the opening of the PR and the comment
files: Dict[str, FileData] # filename -> file data, files before the PR (before the first PR commits)
diffs_before: Dict[str, Diff] # filename -> diff, diffs between the opening of the PR and the comment
comment: str
diffs_after: List[Diff] # changes after the comment
diffs_after: Dict[str, Diff] # filename -> diff, changes after the comment
@dataclass
class Dataset: