aboutsummaryrefslogtreecommitdiffstats
path: root/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql
diff options
context:
space:
mode:
authorrobot-piglet <robot-piglet@yandex-team.com>2024-12-11 11:40:34 +0300
committerVitaly Isaev <vitalyisaev@ydb.tech>2024-12-12 10:12:05 +0000
commit8f4257fa81e2a1752501d30cae4f891cbfd43951 (patch)
tree1d2319c85e123f31139df500094a05de1acdd89a /yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql
parent32d3316b041c1d66d356a924f55f3c585809aee2 (diff)
downloadydb-8f4257fa81e2a1752501d30cae4f891cbfd43951.tar.gz
Intermediate changes
commit_hash:09395197ee7b5fe0fedb7f40787986adc8ba8abb
Diffstat (limited to 'yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql')
-rw-r--r--yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql4
1 files changed, 2 insertions, 2 deletions
diff --git a/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql b/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql
index 3e8637fc54..2143f06bf2 100644
--- a/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql
+++ b/yql/essentials/tests/s-expressions/suites/Optimizers/FuseLMapAfterReduce.yql
@@ -26,7 +26,7 @@
(let inputType (CallableArgumentType (TypeOf (ScriptUdf 'Python3 '"f" (CallableType '() '((StreamType (StructType '('"cnt" (DataType 'Uint64)) '('"pass" (DataType 'Int32)) '('"skey" (DataType 'Uint32))))) '((StreamType (StructType '('"cnt" (DataType 'Uint64)) '('"skey" (DataType 'Uint32))))) '((DataType 'Int32))) (String '@@
def f(input,x):
for i in list(input):
- d = i.__dict__
+ d = {name: getattr(i, name) for name in i.__class__.__match_args__}
d["pass"] = x
yield d
@@))) '0))
@@ -34,7 +34,7 @@ def f(input,x):
(return (Apply (ScriptUdf 'Python3 '"f" (CallableType '() '((StreamType (StructType '('"cnt" (DataType 'Uint64)) '('"pass" (DataType 'Int32)) '('"skey" (DataType 'Uint32))))) '((StreamType (StructType '('"cnt" (DataType 'Uint64)) '('"skey" (DataType 'Uint32))))) '((DataType 'Int32))) (String '@@
def f(input,x):
for i in list(input):
- d = i.__dict__
+ d = {name: getattr(i, name) for name in i.__class__.__match_args__}
d["pass"] = x
yield d
@@)) inputRowsList (Int32 '"1")))