codeclimate

Kev1CO · Jul 15, 2024 · 5639923 · 5639923
1 parent 4656bb0
commit 5639923
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 151 deletions.
diff --git a/...fest/examples/dynamics/reaching_task/reaching_task_pulse_duration_optimization_testing.py b/...fest/examples/dynamics/reaching_task/reaching_task_pulse_duration_optimization_testing.py
diff --git a/cocofest/examples/dynamics/reaching_task/visualization/pulse_duration_subplot.py b/cocofest/examples/dynamics/reaching_task/visualization/pulse_duration_subplot.py
@@ -25,13 +25,11 @@
 nb_stim = len(data_minimize_fatigue["parameters"][pulse_duration_keys[0]])
 width = round(data_minimize_fatigue["time"][-1], 2) / nb_stim
 
-pw_for_force_optim = [
-    data_minimize_force["parameters"][pulse_duration_keys[i]] * 1000000 for i in range(len(pulse_duration_keys))
-]
-pw_for_fatigue_optim = [
-    data_minimize_fatigue["parameters"][pulse_duration_keys[i]] * 1000000 for i in range(len(pulse_duration_keys))
-]
-pw_list = [pw_for_force_optim, pw_for_fatigue_optim]
+pw_data_list = [data_minimize_force["parameters"], data_minimize_fatigue["parameters"]]
+
+pw_list = []
+for j in range(2):
+    pw_list.append([pw_data_list[j][pulse_duration_keys[i]] * 1000000 for i in range(len(pulse_duration_keys))])
 
 plasma = cm = plt.get_cmap("plasma")
 cNorm = colors.Normalize(vmin=100, vmax=600)
@@ -57,7 +55,7 @@ def plot_graph(datas):
             axs[i].set_xlabel("Time (s)")
 
         for j in range(nb_stim):
-            value = pw_for_force_optim[i][j]
+            value = datas[i][j]
 
             color = scalarMap.to_rgba(value)
             axs[i].barh(muscle_names[i], width, left=j * width, height=0.5, color=color)