Update MasteryParseTreeWalker.java (#2482)

finos · Dec 1, 2023 · f7c44ec · f7c44ec
1 parent 3ffb11f
commit f7c44ec
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...rg/finos/legend/engine/language/pure/dsl/mastery/grammar/from/MasteryParseTreeWalker.java b/...rg/finos/legend/engine/language/pure/dsl/mastery/grammar/from/MasteryParseTreeWalker.java
@@ -378,13 +378,13 @@ private Lambda visitLambdaWithFilter(String propertyName, MasteryParserGrammar.C
         String inputFilter = ctx.getText().replaceAll(DOLLAR_SIGN, INPUT);
         return domainParser.parseLambda(
                 format(PRECEDENCE_LAMBDA_WITH_FILTER, propertyName, inputFilter),
-                "", 0, 0, true);
+                "", 0, 0, walkerSourceInformation.getReturnSourceInfo());
     }
 
     private Lambda visitLambdaWithoutFilter(String propertyName)
     {
         return domainParser.parseLambda(format(SIMPLE_PRECEDENCE_LAMBDA, propertyName),
-                "", 0, 0, true);
+                "", 0, 0, walkerSourceInformation.getReturnSourceInfo());
     }
 
     private RuleScope visitRuleScopeWithoutPrecedence(MasteryParserGrammar.ScopeContext ctx, Set<String> uniqueScopes,PrecedenceRule precedenceRule)
@@ -748,7 +748,7 @@ private List<Lambda> visitQueryExpressions(MasteryParserGrammar.QueryExpressions
 
     private Lambda visitLambda(MasteryParserGrammar.LambdaFunctionContext ctx)
     {
-        return domainParser.parseLambda(ctx.getText(), "", 0, 0, true);
+        return domainParser.parseLambda(ctx.getText(), "", 0, 0, walkerSourceInformation.getReturnSourceInfo());
     }
 
     private ResolutionKeyType visitResolutionKeyType(MasteryParserGrammar.ResolutionQueryKeyTypeContext ctx)