aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorzverevgeny <zverevgeny@ydb.tech>2023-08-22 20:56:07 +0300
committerzverevgeny <zverevgeny@ydb.tech>2023-08-22 22:04:55 +0300
commit16b49a7ff6a0740a19bab5650934824d3d7030f4 (patch)
tree0658e582336b9214468cb6ff6538ae36bcd4a1af
parente4a985d19c86ab1131bbb93cc24a83132449653b (diff)
downloadydb-16b49a7ff6a0740a19bab5650934824d3d7030f4.tar.gz
YQL-16375 Fix rewrite ShuffleByKeys with PartitionsByKeys for non-streams
-rw-r--r--ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp8
1 files changed, 5 insertions, 3 deletions
diff --git a/ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp b/ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp
index 83c482b49e..5eb55995c9 100644
--- a/ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp
+++ b/ydb/library/yql/providers/yt/provider/yql_yt_logical_optimize.cpp
@@ -2645,9 +2645,11 @@ protected:
.ListHandlerLambda<TCoLambda>()
.Args({ TStringBuf("stream") })
.Body<TCoForwardList>()
- .Stream<TExprApplier>()
- .Apply(shuffle.ListHandlerLambda().Body())
- .With(shuffle.ListHandlerLambda().Args().Arg(0), TStringBuf("stream"))
+ .Stream<TCoToStream>()
+ .Input<TExprApplier>()
+ .Apply(shuffle.ListHandlerLambda().Body())
+ .With(shuffle.ListHandlerLambda().Args().Arg(0), TStringBuf("stream"))
+ .Build()
.Build()
.Build()
.Build()