From 04a15b1b1ddf089a467399027493e734a87e4b74 Mon Sep 17 00:00:00 2001
From: fousekjan <jan.fousek@univ-amu.fr>
Date: Mon, 7 Jun 2021 17:06:30 +0200
Subject: [PATCH] wip
---
phase_plane.py | 14 ++++++++++++--
1 file changed, 12 insertions(+), 2 deletions(-)
diff --git a/phase_plane.py b/phase_plane.py
index fac0fea..7231498 100644
--- a/phase_plane.py
+++ b/phase_plane.py
@@ -3,7 +3,7 @@ from ipywidgets import interact, FloatSlider, Dropdown, ToggleButton
import numpy as np
import matplotlib.pylab as plt
import matplotlib.gridspec as gridspec
-
+from tvb.simulator.lab import integrators
def phase_plane_interactive(model, integrator):
@@ -101,6 +101,16 @@ def phase_plane_interactive(model, integrator):
if param_kwargs['trajectory']:
+ if isinstance(integrator, integrators.IntegratorStochastic):
+ if integrator.noise.ntau > 0.0:
+ integrator.noise.configure_coloured(integrator.dt,
+ (1, model.nvar, 1,
+ model.number_of_modes))
+ else:
+ integrator.noise.configure_white(integrator.dt,
+ (1, model.nvar, 1,
+ model.number_of_modes))
+
svx_ind = model.state_variables.index(svx)
svy_ind = model.state_variables.index(svy)
@@ -142,7 +152,7 @@ def phase_plane_interactive(model, integrator):
param_range = param_def.domain
if param_range is None:
continue
- param_value = getattr(model, param_name)[0]
+ param_value = getattr(model, param_name).item()
param_kwargs[param_name] = FloatSlider(
min=param_range.lo, max=param_range.hi, value=param_value)
--
GitLab