diff --git a/nodes/__init__.py b/nodes/__init__.py index 0a7b59b..66bd10c 100644 --- a/nodes/__init__.py +++ b/nodes/__init__.py @@ -209,6 +209,11 @@ class ARDrawSignal: } } +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,)