From 9982e2bcc93d67891a97136e2b00750184f1bdc2 Mon Sep 17 00:00:00 2001
From: Karma Riuk <riukkarma@gmail.com>
Date: Fri, 14 Mar 2025 12:05:39 +0100
Subject: [PATCH] added the diff class to dataset entries

---
 dataset.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/dataset.py b/dataset.py
index 3b150c1..a1f1a52 100644
--- a/dataset.py
+++ b/dataset.py
@@ -16,13 +16,18 @@ class Metadata:
     reason_for_failure: str = ""
     last_cmd_error_msg: str = ""
 
+@dataclass
+class Diff:
+    filename: str
+    patch: str
+
 @dataclass
 class DatasetEntry:
     metadata: Metadata
     files: List[FileData] # files before the PR (before the first PR commits)
-    diffs_before: List[str] # diffs between the opening of the PR and the comment
+    diffs_before: List[Diff] # diffs between the opening of the PR and the comment
     comment: str
-    diffs_after: List[str] # changes after the comment
+    diffs_after: List[Diff] # changes after the comment
 
 @dataclass
 class Dataset: