Prefer dataclass

aai-institute · Dec 13, 2024 · 523e5b8 · 523e5b8
1 parent b00a132
commit 523e5b8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/pydvl/valuation/dataset.py b/src/pydvl/valuation/dataset.py
@@ -42,7 +42,8 @@
 
 import logging
 from collections import OrderedDict
-from typing import Any, NamedTuple, Sequence
+from dataclasses import dataclass
+from typing import Any, Sequence
 
 import numpy as np
 from numpy.typing import NDArray
@@ -55,7 +56,8 @@
 logger = logging.getLogger(__name__)
 
 
-class RawData(NamedTuple):
+@dataclass(frozen=True)
+class RawData:
     x: NDArray
     y: NDArray