From 107fd8e8e6cc0476974501221441a059c43de283 Mon Sep 17 00:00:00 2001 From: cwinkler Date: Thu, 27 Feb 2025 16:46:51 +0100 Subject: [PATCH] fix dataset identifiers --- aviary/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aviary/utils.py b/aviary/utils.py index 6da411b9..c1c7d80d 100644 --- a/aviary/utils.py +++ b/aviary/utils.py @@ -536,7 +536,7 @@ def results_multitask( # TODO cleaner way to get identifier names if save_results: save_results_dict( - dict(zip(test_loader.dataset.identifiers, *ids)), + dict(zip(test_loader.dataset.dataset.identifiers, *ids)), results_dict, model_name, f"-r{run_id}",