Code enhancements (#179)

- Use method references.
wala · Apr 9, 2024 · 5dbf07c · 5dbf07c
1 parent eb59763
commit 5dbf07c
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/...m.wala.cast.python.jython3/source/com/ibm/wala/cast/python/parser/PythonModuleParser.java b/...m.wala.cast.python.jython3/source/com/ibm/wala/cast/python/parser/PythonModuleParser.java
@@ -36,6 +36,8 @@
 import org.antlr.runtime.ANTLRInputStream;
 import org.antlr.runtime.CharStream;
 import org.python.antlr.ast.ImportFrom;
+import org.python.antlr.ast.Name;
+import org.python.antlr.ast.alias;
 
 public class PythonModuleParser extends PythonParser<ModuleEntry> {
 
@@ -63,10 +65,7 @@ protected PythonParser<ModuleEntry>.CAstVisitor makeVisitor(
       public CAstNode visitImportFrom(ImportFrom importFrom) throws Exception {
         Optional<String> s =
             importFrom.getInternalModuleNames().stream()
-                .map(
-                    n -> {
-                      return n.getInternalId();
-                    })
+                .map(Name::getInternalId)
                 .reduce(
                     (a, b) -> {
                       return a + "/" + b;
@@ -89,7 +88,7 @@ public CAstNode visitImportFrom(ImportFrom importFrom) throws Exception {
             return Ast.makeNode(
                 CAstNode.BLOCK_STMT,
                 importFrom.getInternalNames().stream()
-                    .map(a -> a.getInternalName())
+                    .map(alias::getInternalName)
                     .map(
                         n ->
                             Ast.makeNode(