From 3d9b386f2217c5fbf616475dd9cf72d83b3ca43b Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Mon, 17 Mar 2025 15:26:45 +0100 Subject: [PATCH] changed the structure of the dataset --- dataset.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/dataset.py b/dataset.py index 5f37c45..7e1a43f 100644 --- a/dataset.py +++ b/dataset.py @@ -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: