fix annotation call

Author: geooo109

sqlglot/optimizer/annotate_types.py

@@ -477,10 +477,12 @@ def _annotate_unary(self, expression: E) -> E:
         else:
             self._set_type(expression, expression.this.type)
 
-        if this_type := expression.this.type:
-            nonnull = this_type.args.get("nonnull")
-            if nonnull and (expr_type := expression.type):
-                expr_type.set("nonnull", nonnull)
+        if (
+            (this_type := expression.this.type)
+            and this_type.args.get("nonnull")
+            and (expr_type := expression.type)
+        ):
+            expr_type.set("nonnull", True)
 
         return expression
 

sqlglot/typing/__init__.py

@@ -21,11 +21,7 @@
     },
     **{
         expr_type: {"annotator": lambda self, e: self._annotate_unary(e)}
-        for expr_type in subclasses(exp.__name__, (exp.Unary, exp.Alias), {exp.Not})
-    },
-    **{
-        expr_type: {"returns": exp.DataType.Type.BOOLEAN}
-        for expr_type in subclasses(exp.__name__, (exp.Connector, exp.Predicate))
+        for expr_type in subclasses(exp.__name__, (exp.Unary, exp.Alias))
     },
     **{
         expr_type: {"returns": exp.DataType.Type.BIGINT}
@@ -58,7 +54,6 @@
             exp.In,
             exp.LogicalAnd,
             exp.LogicalOr,
-            exp.Not,
             exp.RegexpLike,
             exp.StartsWith,
         }