Update __init__.py

2024-01-01 17:42:31 -08:00
parent 7322bb0bb6
commit 306131952b
1 changed files with 5 additions and 1 deletions
@@ -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,)