From dbecfb39742c1d01f39bfaec4f83b0f263753659 Mon Sep 17 00:00:00 2001
From: jasperalbers <jasperalbers@posteo.de>
Date: Wed, 19 Jan 2022 13:17:33 +0100
Subject: [PATCH] update use of rng seed

---
 multiarea_model/default_params.py |  2 +-
 multiarea_model/simulation.py     | 19 +++++++++++--------
 2 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/multiarea_model/default_params.py b/multiarea_model/default_params.py
index 499bfae..c34b2da 100644
--- a/multiarea_model/default_params.py
+++ b/multiarea_model/default_params.py
@@ -38,7 +38,7 @@ Simulation parameters
 """
 sim_params = {
     # master seed for random number generators
-    'master_seed': 0,
+    'rng_seed': 0,
     # simulation step (in ms)
     'dt': 0.1,
     # simulated time (in ms)
diff --git a/multiarea_model/simulation.py b/multiarea_model/simulation.py
index 610c317..a3f1e68 100644
--- a/multiarea_model/simulation.py
+++ b/multiarea_model/simulation.py
@@ -145,7 +145,7 @@ class Simulation:
         Prepare NEST Kernel.
         """
         nest.ResetKernel()
-        master_seed = self.params['master_seed']
+        rng_seed = self.params['rng_seed']
         num_processes = self.params['num_processes']
         local_num_threads = self.params['local_num_threads']
         vp = num_processes * local_num_threads
@@ -153,15 +153,18 @@ class Simulation:
                               'total_num_virtual_procs': vp,
                               'overwrite_files': True,
                               'data_path': os.path.join(self.data_dir, 'recordings'),
-                              'print_time': False,
-                              'grng_seed': master_seed,
-                              'rng_seeds': list(range(master_seed + 1,
-                                                      master_seed + vp + 1))})
-
+                              'print_time': False})
+        if self.network.params['USING_NEST_3']:
+            nest.SetKernelStatus({'rng_seed': rng_seed})
+        else:
+            nest.SetKernelStatus({'grng_seed': rng_seed,
+                                  'rng_seeds': list(range(rng_seed + 1,
+                                                          rng_seed + vp + 1))})
+            self.pyrngs = [np.random.RandomState(s) for s in list(range(
+                rng_seed + vp + 1, rng_seed + 2 * (vp + 1)))]
         nest.SetDefaults(self.network.params['neuron_params']['neuron_model'],
                          self.network.params['neuron_params']['single_neuron_dict'])
-        self.pyrngs = [np.random.RandomState(s) for s in list(range(
-            master_seed + vp + 1, master_seed + 2 * (vp + 1)))]
+
 
     def create_recording_devices(self):
         """
-- 
GitLab