update multi sims to use times vector

neurodsp-tools · Nov 26, 2024 · e1f186f · e1f186f
1 parent 7d52d12
commit e1f186f
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/tutorials/sim/plot_07_SimMulti.py b/tutorials/sim/plot_07_SimMulti.py
@@ -11,6 +11,7 @@
                                 sim_across_values, sim_multi_across_values)
 from neurodsp.sim.update import create_updater, create_sampler, ParamSampler
 from neurodsp.plts.time_series import plot_time_series, plot_multi_time_series
+from neurodsp.utils.data import create_times
 
 ###################################################################################################
 # Simulate Multiple Signals Together
@@ -41,8 +42,11 @@
 
 ###################################################################################################
 
+# Create a times definition corresponding to the simulations
+times = create_times(params['n_seconds'], params['fs'])
+
 # Plot the simulated signals
-plot_multi_time_series(None, sigs)
+plot_multi_time_series(times, sigs)
 
 ###################################################################################################
 # SigIter
@@ -65,7 +69,7 @@
 
 # Iterate with the object to create simulations
 for tsig in sig_iter:
-    plot_time_series(None, tsig)
+    plot_time_series(times, tsig)
 
 ###################################################################################################
 # Simulate From Sampler
@@ -108,7 +112,7 @@
 ###################################################################################################
 
 # Plot the set of sampled simulations
-plot_multi_time_series(None, sampled_sims)
+plot_multi_time_series(times, sampled_sims)
 
 ###################################################################################################
 # Simulate Across Values
@@ -140,7 +144,8 @@
 
 ###################################################################################################
 
-plot_multi_time_series(None, sims_across_params)
+# Plot the simulated time series from sampled parameters
+plot_multi_time_series(times, sims_across_params)
 
 ###################################################################################################
 # Simulate Multiple Instances Across Values