diff --git a/src/meegsim/_check.py b/src/meegsim/_check.py index 17c59cb..ddd6d04 100644 --- a/src/meegsim/_check.py +++ b/src/meegsim/_check.py @@ -352,7 +352,7 @@ def check_extents(extents, n_sources): for extent in extents: if extent is not None: # Check if each extent is a number - if not isinstance(extent, (int, float)): + if not isinstance(extent, (int, float, np.integer, np.floating)): raise ValueError(f"Extent {extent} must be a number.") # Check if each extent is positive diff --git a/src/meegsim/simulate.py b/src/meegsim/simulate.py index 7b5a798..fd6a404 100644 --- a/src/meegsim/simulate.py +++ b/src/meegsim/simulate.py @@ -176,6 +176,10 @@ def add_patch_sources( self._source_groups.append(patch_sg) self._sources.extend(patch_sg.names) + # Check if SNR should be adjusted + if patch_sg.snr is not None: + self.is_snr_adjusted = True + # Return the names of newly added sources return patch_sg.names