From 4bd54fe46a9713745185b1f60253e1d96fa1fa7a Mon Sep 17 00:00:00 2001 From: David Marx Date: Mon, 1 Jan 2024 17:06:39 -0800 Subject: [PATCH] Update __init__.py --- nodes/__init__.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/nodes/__init__.py b/nodes/__init__.py index 66bd10c..f43a6bf 100644 --- a/nodes/__init__.py +++ b/nodes/__init__.py @@ -209,11 +209,12 @@ class ARDrawSignal: } } + def main(self, signal): + img_tensor = draw_signal(signal['y'], signal['sr']) + return (img_tensor,) + + NODE_CLASS_MAPPINGS["ARDrawSignal"] = ARDrawSignal NODE_DISPLAY_NAME_MAPPINGS["ARDrawSignal"] = "Draw Audio Signal" ############################################ - - def main(self, signal): - img_tensor = draw_signal(signal['y'], signal['sr']) - return (img_tensor,)