Skip to content

Commit

Permalink
remove some predict and metric function from mapsManager
Browse files Browse the repository at this point in the history
  • Loading branch information
camillebrianceau committed Aug 28, 2024
1 parent 3a1e5d4 commit 6f23958
Showing 1 changed file with 13 additions and 3 deletions.
16 changes: 13 additions & 3 deletions tests/test_predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,9 @@

import pytest

from clinicadl.metrics.utils import get_metrics
from clinicadl.predict.predict_manager import PredictManager
from clinicadl.predict.utils import get_prediction

from .testing_tools import compare_folders, modify_maps

Expand Down Expand Up @@ -115,10 +117,18 @@ def test_predict(cmdopt, tmp_path, test_name):
predict_manager.predict()

for mode in modes:
predict_manager.maps_manager.get_prediction(data_group="test-RANDOM", mode=mode)
get_prediction(
predict_manager.maps_manager.maps_path,
predict_manager.maps_manager.split_name,
data_group="test-RANDOM",
mode=mode,
)
if use_labels:
predict_manager.maps_manager.get_metrics(
data_group="test-RANDOM", mode=mode
get_metrics(
predict_manager.maps_manager.maps_path,
predict_manager.maps_manager.split_name,
data_group="test-RANDOM",
mode=mode,
)

assert compare_folders(
Expand Down

0 comments on commit 6f23958

Please sign in to comment.