diff --git a/recording_generation.py b/recording_generation.py index cb4b329..154aaf2 100644 --- a/recording_generation.py +++ b/recording_generation.py @@ -60,9 +60,7 @@ class RecordingGenerator: def generate_wb(self, num: int = 2, length: int = 8192): for i in range(num): - recording = create_noise_recording( - length=length, rms_power=0.2, seed=num - ) + recording = create_noise_recording(length=length, rms_power=0.2, seed=num) recording.to_npy(filename=f"wb{i + 1}", overwrite=True) print(f"wb{i + 1} file saved.") diff --git a/signal_generation.py b/signal_generation.py index 2fbf120..1c606e5 100644 --- a/signal_generation.py +++ b/signal_generation.py @@ -97,15 +97,13 @@ def create_lfm_recording( def create_noise_recording( - rms_power: float, - length: int, - seed: int | None = None + rms_power: float, length: int, seed: int | None = None ) -> Recording: """Generate a Recording of Additive White Gaussian Noise (AWGN).""" # 1. Create a repeating pseudo-random envelope if seed is not None: np.random.seed(seed) - + # 2. Sigma for complex AWGN: sigma = np.sqrt(rms_power / 2)