Update __init__.py

2024-01-01 17:06:39 -08:00
parent 6108853fe9
commit 4bd54fe46a
1 changed files with 5 additions and 4 deletions
@@ -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,)