feat(skore-hub-project): Remove media from CrossValidationReportPayload that are not used by the hub

Author: thomass-dev

skore-hub-project/src/skore_hub_project/report/cross_validation_report.py

@@ -12,12 +12,6 @@
 from skore_hub_project.artefact import CrossValidationReportArtefact
 from skore_hub_project.media import (
     EstimatorHtmlRepr,
-    PrecisionRecallTest,
-    PrecisionRecallTrain,
-    PredictionErrorTest,
-    PredictionErrorTrain,
-    RocTest,
-    RocTrain,
 )
 from skore_hub_project.media.media import Media
 from skore_hub_project.metric import (
@@ -129,18 +123,7 @@ class CrossValidationReportPayload(ReportPayload):
             PredictTimeTrainStd,
         ),
     )
-    MEDIAS: ClassVar[tuple[Media, ...]] = cast(
-        tuple[Media, ...],
-        (
-            EstimatorHtmlRepr,
-            PrecisionRecallTest,
-            PrecisionRecallTrain,
-            PredictionErrorTest,
-            PredictionErrorTrain,
-            RocTest,
-            RocTrain,
-        ),
-    )
+    MEDIAS: ClassVar[tuple[Media, ...]] = cast(tuple[Media, ...], (EstimatorHtmlRepr,))
 
     report: CrossValidationReport = Field(repr=False, exclude=True)
 

skore-hub-project/tests/unit/report/test_cross_validation_report.py

@@ -9,10 +9,6 @@
 from skore_hub_project.artefact.serializer import Serializer
 from skore_hub_project.media import (
     EstimatorHtmlRepr,
-    PrecisionRecallTest,
-    PrecisionRecallTrain,
-    RocTest,
-    RocTrain,
 )
 from skore_hub_project.metric import (
     AccuracyTestMean,
@@ -202,13 +198,7 @@ def test_metrics(self, payload):
         ]
 
     def test_related_items(self, payload):
-        assert list(map(type, payload.related_items)) == [
-            EstimatorHtmlRepr,
-            PrecisionRecallTest,
-            PrecisionRecallTrain,
-            RocTest,
-            RocTrain,
-        ]
+        assert list(map(type, payload.related_items)) == [EstimatorHtmlRepr]
 
     @mark.usefixtures("monkeypatch_routes")
     def test_model_dump(self, small_cv_binary_classification, payload):