FIX: compute ArraySize with len() in tnp backend

ComPWA · Jan 11, 2024 · 3185ec8 · 3185ec8
1 parent 2316267
commit 3185ec8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/tensorwaves/function/sympy/_printer.py b/src/tensorwaves/function/sympy/_printer.py
@@ -81,5 +81,6 @@ class TensorflowPrinter(CustomNumPyPrinter):
     def __init__(self) -> None:
         # https://github.com/sympy/sympy/blob/f1384c2/sympy/printing/printer.py#L21-L72
         super().__init__()
+        self.known_functions["ArraySize"] = "len"
         self.known_functions["ComplexSqrt"] = "sqrt"
         self.printmethod = "_tensorflow_code"