From e0fc7aeed43322a9c8b7367dbcbaf84364b0c582 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Beno=C3=AEt=20Sierro?= <benoit.sierro@bluewin.ch>
Date: Tue, 12 Sep 2023 15:39:33 +0200
Subject: [PATCH] new: Rng customization in NoiseMeasurement

---
 src/scgenerator/noise.py   | 3 ++-
 src/scgenerator/spectra.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/scgenerator/noise.py b/src/scgenerator/noise.py
index feee016..e8475ca 100644
--- a/src/scgenerator/noise.py
+++ b/src/scgenerator/noise.py
@@ -16,6 +16,7 @@ class NoiseMeasurement:
     psd: np.ndarray
     phase: np.ndarray | None = None
     psd_interp: interp1d = field(init=False)
+    rng: np.random.Generator = field(default_factory=np.random.default_rng)
     _window_functions: ClassVar[dict[str, Callable[[int], np.ndarray]]] = {}
 
     @classmethod
@@ -190,7 +191,7 @@ class NoiseMeasurement:
 
         freq, spec = self.sample_spectrum(nt, dt)
         if phase is None:
-            phase = 2 * np.pi * np.random.rand(len(freq))
+            phase = 2 * np.pi * self.rng.random(len(freq))
         time, dt = math.irfftfreq(freq, True)
 
         amp = np.sqrt(spec) * np.exp(1j * phase)
diff --git a/src/scgenerator/spectra.py b/src/scgenerator/spectra.py
index fd28f37..e2bda18 100644
--- a/src/scgenerator/spectra.py
+++ b/src/scgenerator/spectra.py
@@ -332,7 +332,7 @@ def _bundle_external_files(params: Parameters, io: PropagationIOHandler):
     existing_files = set(io.keys())
     for _, value in params.items():
         if isinstance(value, DataFile):
-            data = value.load_data()
+            data = value.load_bytes()
 
             value.io = io
             value.prefix = "zip"