fix bugs of random generator setting

zhiim · Jun 7, 2024 · 4073cbe · 4073cbe
1 parent 79c74c5
commit 4073cbe
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/classical_doa/__init__.py b/classical_doa/__init__.py
@@ -1,2 +1,2 @@
-__version__ = "1.0.1"
-__author__ = 'Qian Xu'
+__version__ = "1.0.2"
+__author__ = "Qian Xu"
diff --git a/classical_doa/arrays.py b/classical_doa/arrays.py
@@ -21,6 +21,8 @@ def __init__(
 
         if rng is None:
             self._rng = np.random.default_rng()
+        else:
+            self._rng = rng
 
     @property
     def num_antennas(self):

diff --git a/classical_doa/signals.py b/classical_doa/signals.py
@@ -16,6 +16,8 @@ def __init__(self, nsamples, fs, rng=None):
 
         if rng is None:
             self._rng = np.random.default_rng()
+        else:
+            self._rng = rng
 
     @property
     def fs(self):