From 6108853fe95ef749ef880c5a989aacece4761b67 Mon Sep 17 00:00:00 2001
From: David Marx <david.marx84@gmail.com>
Date: Mon, 1 Jan 2024 17:05:18 -0800
Subject: [PATCH] Update __init__.py

---
 nodes/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

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,)