style: linting

Aarhus-Psychiatry-Research · Feb 16, 2023 · 10b6c51 · 10b6c51
1 parent 8e92d94
commit 10b6c51
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/psycop_model_training/model_eval/base_artifacts/plots/precision_recall.py b/src/psycop_model_training/model_eval/base_artifacts/plots/precision_recall.py
@@ -28,11 +28,13 @@ def plot_precision_recall(
         Union[None, Path]: None if save_path is None, else path to saved figure.
     """
     precision, recall, _ = precision_recall_curve(
-        y_true=eval_dataset.y, probas_pred=eval_dataset.y_hat_probs
+        y_true=eval_dataset.y,
+        probas_pred=eval_dataset.y_hat_probs,
     )
 
     auprc = average_precision_score(
-        y_true=eval_dataset.y, y_score=eval_dataset.y_hat_probs
+        y_true=eval_dataset.y,
+        y_score=eval_dataset.y_hat_probs,
     )
 
     legend_label = "AUPRC = "

diff --git a/tests/model_evaluation/test_visualizations.py b/tests/model_evaluation/test_visualizations.py
@@ -186,5 +186,6 @@ def test_plot_roc_auc(synth_eval_dataset: EvalDataset):
 
 def test_plot_precision_recall(synth_eval_dataset: EvalDataset):
     plot_precision_recall(
-        eval_dataset=synth_eval_dataset, save_path="tmp/test_plot_precision_recall.png"
+        eval_dataset=synth_eval_dataset,
+        save_path="tmp/test_plot_precision_recall.png",
     )