make guard more specific

ydb-platform · Oct 21, 2024 · a83c9d6 · a83c9d6
1 parent 6e0d92e
commit a83c9d6
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/ydb/library/yql/dq/opt/dq_opt_phy.cpp b/ydb/library/yql/dq/opt/dq_opt_phy.cpp
@@ -2734,12 +2734,10 @@ TExprBase DqPropagatePrecomuteTake(TExprBase node, TExprContext& ctx, IOptimizat
     }
 
     auto* typeAnn = precompute.Connection().Raw()->GetTypeAnn();
-    if (!typeAnn || typeAnn->GetKind() != ETypeAnnotationKind::Stream) {
-        return node;
-    }
 
-    typeAnn = typeAnn->Cast<TStreamExprType>()->GetItemType();
-    if (typeAnn->GetKind() != ETypeAnnotationKind::List && typeAnn->GetKind() != ETypeAnnotationKind::Optional) {
+    YQL_ENSURE(typeAnn && typeAnn->GetKind() == ETypeAnnotationKind::List);
+    typeAnn = typeAnn->Cast<TListExprType>()->GetItemType();
+    if (typeAnn->GetKind() == ETypeAnnotationKind::Struct) {
         return node;
     }