remove self in task

aramis-lab · Sep 3, 2024 · 94121f1 · 94121f1
1 parent 8bdca0f
commit 94121f1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/clinicadl/trainer/task_manager.py b/clinicadl/trainer/task_manager.py
@@ -233,7 +233,9 @@ def test(
                     data,
                     outputs.float(),
                 )
-                row_df = pd.DataFrame(row, columns=self.columns)
+                row_df = pd.DataFrame(
+                    row, columns=columns(network_task, mode, n_classes)
+                )
                 results_df = pd.concat([results_df, row_df])
 
             del outputs, loss_dict