From 306131952b885273efdca8bae33ff4cf63f1857f Mon Sep 17 00:00:00 2001 From: David Marx Date: Mon, 1 Jan 2024 17:42:31 -0800 Subject: [PATCH] Update __init__.py --- nodes/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/nodes/__init__.py b/nodes/__init__.py index a9abebf..c491efc 100644 --- a/nodes/__init__.py +++ b/nodes/__init__.py @@ -211,7 +211,11 @@ class ARDrawSignal: } def main(self, signal): - img_tensor = draw_signal(signal['y'], signal['sr']) + img_tensor = draw_signal( + y=signal['y'], + sr=signal['sr'], + raw=signal.get('is_raw',True), + ) return (img_tensor,)