Improve conversion of typenames to types

substrait-io · May 8, 2024 · b90d55d · b90d55d
1 parent e9508a4
commit b90d55d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/substrait/sql/functions_catalog.py b/src/substrait/sql/functions_catalog.py
@@ -47,6 +47,10 @@ def arguments_type(self) -> list[SubstraitType | None]:
         return [self._type_from_name(arg) for arg in self.arguments]
 
     def _type_from_name(self, typename: str) -> SubstraitType | None:
+        # TODO: improve support complext type like LIST?<any>
+        typename, *_ = typename.split("<", 1)
+        typename = typename.lower()
+
         nullable = False
         if typename.endswith("?"):
             nullable = True