YQL-14762: dqReplicate lambda's return type may be stream

ref:a50a6ef776b9c9b8d9a50541c97e8f86b46fd256

1 files changed, 2 insertions, 2 deletions

diff --git a/ydb/library/yql/dq/type_ann/dq_type_ann.cpp b/ydb/library/yql/dq/type_ann/dq_type_ann.cpp
index 75d76dd52e..aea96d65b2 100644
--- a/ydb/library/yql/dq/type_ann/dq_type_ann.cpp
+++ b/ydb/library/yql/dq/type_ann/dq_type_ann.cpp
@@ -683,10 +683,10 @@ TStatus AnnotateDqReplicate(const TExprNode::TPtr& input, TExprContext& ctx) {
         if (!lambda->GetTypeAnn()) {
             return TStatus::Repeat;
         }
-        if (!EnsureFlowType(lambda->Pos(), *lambda->GetTypeAnn(), ctx)) {
+        const TTypeAnnotationNode* lambdaItemType = nullptr;
+        if (!EnsureNewSeqType<false, false>(*lambda, ctx, &lambdaItemType)) {
             return TStatus::Error;
         }
-        auto lambdaItemType = lambda->GetTypeAnn()->Cast<TFlowExprType>()->GetItemType();
         if (!EnsurePersistableType(lambda->Pos(), *lambdaItemType, ctx)) {
             return TStatus::Error;
         }