Fixing AWGN

TorchDSP · Sep 23, 2023 · 8c792b8 · 8c792b8
1 parent b4c17f6
commit 8c792b8
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 288 deletions.
diff --git a/docs/notebooks/.gitkeep b/docs/notebooks/.gitkeep
diff --git a/docs/notebooks/00_sig53_dataset.ipynb b/docs/notebooks/00_sig53_dataset.ipynb
diff --git a/docs/notebooks/sig53/.gitkeep b/docs/notebooks/sig53/.gitkeep
diff --git a/torchsig/transforms/transforms.py b/torchsig/transforms/transforms.py
@@ -891,7 +891,7 @@ def __call__(self, data: Any) -> Any:
             return new_data
 
         else:
-            noise_power_db = self.noise_power_db(size=data.shape[0])
+            noise_power_db = self.noise_power_db(num=data.shape[0])
             noise_power_db = (
                 10 * np.log10(noise_power_db) if self.linear else noise_power_db
             )