From ad5aa22ae6283674fe6142969e99c810fc46aa39 Mon Sep 17 00:00:00 2001
From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com>
Date: Tue, 30 Apr 2024 03:35:30 +0200
Subject: [PATCH] Add missing `import` for `saveNpArray` and add a missing
 `f`-string

---
 datasets/raise/split_and_compare_prnus_of_subgroups.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py
index 6fc5e44..5e385bb 100755
--- a/datasets/raise/split_and_compare_prnus_of_subgroups.py
+++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py
@@ -3,7 +3,7 @@
 import numpy as np
 import matplotlib.pyplot as plt
 from tqdm import tqdm
-from utils import denoise, iterativeMean, getColorChannel, escapeFilePath, Color, mergeSingleColorChannelImagesAccordingToBayerFilter, rescaleRawImageForDenoiser, updateExtremes
+from utils import denoise, iterativeMean, getColorChannel, escapeFilePath, Color, mergeSingleColorChannelImagesAccordingToBayerFilter, rescaleRawImageForDenoiser, updateExtremes, saveNpArray
 import sys
 import os
 from random import shuffle
@@ -67,6 +67,6 @@ plt.title(f'RMS between both subgroups estimated PRNUs with {DENOISER} denoiser
 plt.xlabel('Number of images of each subgroup')
 plt.ylabel('RMS between both subgroups estimated PRNUs')
 plt.plot(rmss)
-saveNpArray('{setting}_rmss', rmss)
+saveNpArray(f'{setting}_rmss', rmss)
 plt.savefig(f'{setting}_rms_between_estimated_prnu_of_2_subgroups.svg')
 #plt.show()