From b0552625f77b1a98cea8d02fe9e6344954ee7f44 Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Wed, 1 Jan 2025 00:24:50 +0900 Subject: [PATCH] chore(format): run black on dev (#104) Co-authored-by: github-actions[bot] --- infer/lib/train/utils.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/infer/lib/train/utils.py b/infer/lib/train/utils.py index 885668b..9f50ca7 100644 --- a/infer/lib/train/utils.py +++ b/infer/lib/train/utils.py @@ -142,7 +142,9 @@ def plot_spectrogram_to_numpy(spectrogram): fig.canvas.draw() try: data = np.array(fig.canvas.renderer.buffer_rgba(), dtype=np.uint8) - data = data.reshape(fig.canvas.get_width_height()[::-1] + (4,))[:, :, :3] # 只取前三个通道（RGB） + data = data.reshape(fig.canvas.get_width_height()[::-1] + (4,))[ + :, :, :3 + ] # 只取前三个通道（RGB） except: data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep="") data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,)) @@ -175,7 +177,9 @@ def plot_alignment_to_numpy(alignment, info=None): fig.canvas.draw() try: data = np.array(fig.canvas.renderer.buffer_rgba(), dtype=np.uint8) - data = data.reshape(fig.canvas.get_width_height()[::-1] + (4,))[:, :, :3] # 只取前三个通道（RGB） + data = data.reshape(fig.canvas.get_width_height()[::-1] + (4,))[ + :, :, :3 + ] # 只取前三个通道（RGB） except: data = np.fromstring(fig.canvas.tostring_rgb(), dtype=np.uint8, sep="") data = data.reshape(fig.canvas.get_width_height()[::-1] + (3,))