Vaggelis feedback

Author: georgesittas

sqlglot/schema.py

@@ -192,7 +192,7 @@ def udf_parts(self, udf: exp.Anonymous) -> t.List[str]:
         # a.b.c(...) is represented as Dot(Dot(a, b), Anonymous(c, ...))
         parent = udf.parent
         parts = [p.name for p in parent.flatten()] if isinstance(parent, exp.Dot) else [udf.name]
-        return list(reversed(parts))
+        return list(reversed(parts))[0 : self.udf_depth()]
 
     def _find_in_trie(
         self,
@@ -253,7 +253,7 @@ def find_udf(self, udf: exp.Anonymous, raise_on_missing: bool = False) -> t.Opti
         Returns:
             The return type of the UDF, or None if not found.
         """
-        parts = self.udf_parts(udf)[0 : self.udf_depth()]
+        parts = self.udf_parts(udf)
         resolved_parts = self._find_in_trie(parts, self.udf_trie, raise_on_missing)
 
         if resolved_parts is None:
@@ -458,7 +458,6 @@ def get_udf_type(
             The return type as a DataType, or UNKNOWN if not found.
         """
         parts = self._normalize_udf(udf, dialect=dialect, normalize=normalize)
-        parts = parts[0 : self.udf_depth()]
         resolved_parts = self._find_in_trie(parts, self.udf_trie, raise_on_missing=False)
 
         if resolved_parts is None: