[rawdataloader] Better typing for specialized RawDataLoaders

src/mednet/libs/classification/data/typing.py

@@ -9,7 +9,7 @@ class ClassificationRawDataLoader(RawDataLoader):
     def __init__(self):
         super().__init__()
 
-    def sample(self, _: typing.Any) -> Sample:
+    def sample(self, _: tuple[str, int, typing.Any | None]) -> Sample:
         """Load whole samples from media.
 
         Parameters

src/mednet/libs/segmentation/data/typing.py

-import typing
-
 from mednet.libs.common.data.typing import RawDataLoader, Sample
 
 
@@ -9,7 +7,7 @@ class SegmentationRawDataLoader(RawDataLoader):
     def __init__(self):
         super().__init__()
 
-    def sample(self, _: typing.Any) -> Sample:
+    def sample(self, _: tuple[str, str, str | None]) -> Sample:
         """Load whole samples from media.
 
         Parameters