Skip to content

Commit

Permalink
docs: minor lint changes
Browse files Browse the repository at this point in the history
  • Loading branch information
Eoghan O'Connell committed Jan 23, 2025
1 parent 1b799f8 commit eb21e3f
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions examples/fft_batch_speeds.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,8 @@
for fft_interface in fft_interfaces:
results = {}
for n_transforms in n_transforms_list:
print(f"Running {n_transforms} transforms with {fft_interface.__name__}")
print(f"Running {n_transforms} transforms with "
f"{fft_interface.__name__}")

# create batches
data_3d = np.repeat(data_3d_prep, repeats=n_transforms, axis=0)
Expand Down Expand Up @@ -74,12 +75,6 @@
color=color, edgecolor='k')
multiplier += 1

# offset = width * multiplier
# rects = ax1.bar(np.array(n_transforms_list) + offset, height=speed_norm, width=width,
# label=fft_interface.__name__)
# ax1.bar_label(rects, padding=3)
# multiplier += 1

ax1.set_xticks(x_pos + (width/2), labels=n_transforms_list)
ax1.set_xlabel("Fourier transform batch size")
ax1.set_ylabel("Time for single FFT [Time / batch size] (s)")
Expand Down

0 comments on commit eb21e3f

Please sign in to comment.