Should not have to pass a 'final' argument

ATOMScience-org · Dec 13, 2024 · 9c8abc3 · 9c8abc3
1 parent 61c18fd
commit 9c8abc3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/atomsci/ddm/pipeline/perf_plots.py b/atomsci/ddm/pipeline/perf_plots.py
@@ -165,8 +165,8 @@ def plot_pred_vs_actual(model, epoch_label='best', threshold=None, error_bars=Fa
         for s, subset in enumerate(subsets):
             perf_data = wrapper.get_perf_data(subset, epoch_label)
             pred_results = perf_data.get_prediction_results()
-            y_actual = perf_data.get_real_values('final')
-            ids, y_pred, y_std = perf_data.get_pred_values('final')
+            y_actual = perf_data.get_real_values()
+            ids, y_pred, y_std = perf_data.get_pred_values()
             r2 = pred_results['r2_score']
             if perf_data.num_tasks > 1:
                 r2_scores = pred_results['task_r2_scores']