From 406027be4049a03acd043b48624fc455faa35bd6 Mon Sep 17 00:00:00 2001
From: Alina Studenova <76951559+astudenova@users.noreply.github.com>
Date: Wed, 9 Oct 2024 12:52:50 +0200
Subject: [PATCH] FIX: enable adjustment of SNR for patch sources (#39)

---
 src/meegsim/_check.py   | 2 +-
 src/meegsim/simulate.py | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/meegsim/_check.py b/src/meegsim/_check.py
index 88efff0..44738e3 100644
--- a/src/meegsim/_check.py
+++ b/src/meegsim/_check.py
@@ -492,7 +492,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 0657e11..7afbb53 100644
--- a/src/meegsim/simulate.py
+++ b/src/meegsim/simulate.py
@@ -180,6 +180,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