reijii/klippain-shaketune-telegramm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixed a None peaks array impossible to iterate

Browse Source
This commit is contained in:
Félix Boisselier
2024-04-24 14:52:54 +02:00
parent f3ed4cd1a9
commit 178fa2ea3b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/graph_creators/graph_vibrations.py
View File

@@ -305,7 +305,7 @@ def plot_global_speed_profile(
y2max = vibration_metric.max() * 1.07 y2max = vibration_metric.max() * 1.07
ax2.set_ylim([y2min, y2max]) ax2.set_ylim([y2min, y2max])
if peaks is not None: if peaks is not None and len(peaks) > 0:
ax2.plot(all_speeds[peaks], vibration_metric[peaks], 'x', color='black', markersize=8, zorder=10) ax2.plot(all_speeds[peaks], vibration_metric[peaks], 'x', color='black', markersize=8, zorder=10)
for idx, peak in enumerate(peaks): for idx, peak in enumerate(peaks):
ax2.annotate( ax2.annotate(
@@ -499,7 +499,7 @@ def plot_vibration_spectrogram(ax, angles, speeds, spectrogram_data, peaks):
) )
# Add peaks lines in the spectrogram to get hint from peaks found in the first graph # Add peaks lines in the spectrogram to get hint from peaks found in the first graph
if peaks is not None: if peaks is not None and len(peaks) > 0:
for idx, peak in enumerate(peaks): for idx, peak in enumerate(peaks):
ax.axvline(speeds[peak], color='cyan', linewidth=0.75) ax.axvline(speeds[peak], color='cyan', linewidth=0.75)
ax.annotate( ax.annotate(